[assets] fix docker image (#8180)

This commit is contained in:
hoshi-hiyouga
2025-05-27 19:01:31 +08:00
committed by GitHub
parent 2bf8e993ab
commit 4d3ffa2ec4
4 changed files with 26 additions and 21 deletions

View File

@@ -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")