From 32a65e89e505fb23951c7afd36644d65b1fdec20 Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Sun, 19 May 2024 18:27:18 +0800
Subject: [PATCH] fix envs

Former-commit-id: 8ee8ac6eba3e2916cf7c6401a0e9d803b3ff88ba
---
 src/llamafactory/extras/misc.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llamafactory/extras/misc.py b/src/llamafactory/extras/misc.py
index 8955acd1..0addf315 100644
--- a/src/llamafactory/extras/misc.py
+++ b/src/llamafactory/extras/misc.py
@@ -225,4 +225,4 @@ def try_download_model_from_ms(model_args: "ModelArguments") -> str:
 
 
 def use_modelscope() -> bool:
-    return bool(int(os.environ.get("USE_MODELSCOPE_HUB", "0")))
+    return os.environ.get("USE_MODELSCOPE_HUB", "0").lower() in ["true", "1"]