sync ms

modules/deepseek.py

@@ -32,17 +32,17 @@ DESCRIPTION = (
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 if device == torch.device("cuda"):
     if EN_US:
-        import modelscope
-
-        MODEL_DIR = modelscope.snapshot_download(MODEL_ID, cache_dir="./__pycache__")
-
-    else:
         import huggingface_hub
 
         MODEL_DIR = huggingface_hub.snapshot_download(
             MODEL_ID, cache_dir="./__pycache__"
         )
 
+    else:
+        import modelscope
+
+        MODEL_DIR = modelscope.snapshot_download(MODEL_ID, cache_dir="./__pycache__")
+
     tokenizer = AutoTokenizer.from_pretrained(MODEL_DIR)
     model = AutoModelForCausalLM.from_pretrained(MODEL_DIR, device_map="auto")