Doven
commited on
Commit
·
bf36e69
1
Parent(s):
544aebe
fix bug
Browse files
workspace/classinput/qwen25llm.py
CHANGED
|
@@ -5,15 +5,7 @@ import os
|
|
| 5 |
model_name = os.path.join(os.path.dirname(__file__), "Qwen25llm")
|
| 6 |
|
| 7 |
|
| 8 |
-
print("
|
| 9 |
-
from huggingface_hub import snapshot_download
|
| 10 |
-
snapshot_download(repo_id="Qwen2.5-3B-Instruct",
|
| 11 |
-
repo_type="model",
|
| 12 |
-
cache_dir=model_name,
|
| 13 |
-
local_dir_use_symlinks=False,
|
| 14 |
-
resume_download=True)
|
| 15 |
-
|
| 16 |
-
|
| 17 |
model = AutoModelForCausalLM.from_pretrained(
|
| 18 |
model_name,
|
| 19 |
torch_dtype="auto",
|
|
|
|
| 5 |
model_name = os.path.join(os.path.dirname(__file__), "Qwen25llm")
|
| 6 |
|
| 7 |
|
| 8 |
+
print("Loading Qwen2.5 files...")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
model = AutoModelForCausalLM.from_pretrained(
|
| 10 |
model_name,
|
| 11 |
torch_dtype="auto",
|