Doven commited on
Commit
bf36e69
·
1 Parent(s): 544aebe
Files changed (1) hide show
  1. workspace/classinput/qwen25llm.py +1 -9
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("Downloading Qwen2.5 files...")
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",