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",
|