[assets] fix docker image (#8180)
This commit is contained in:
@@ -15,6 +15,7 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
import os
|
||||
import platform
|
||||
|
||||
import accelerate
|
||||
@@ -83,4 +84,9 @@ def print_env() -> None:
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
if os.path.exists("data"):
|
||||
info["Default data directory"] = "detected"
|
||||
else:
|
||||
info["Default data directory"] = "not detected"
|
||||
|
||||
print("\n" + "\n".join([f"- {key}: {value}" for key, value in info.items()]) + "\n")
|
||||
|
||||
Reference in New Issue
Block a user