likewendy commited on
Commit
fd944cc
·
1 Parent(s): b6f8459
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,12 +10,12 @@ os.system('huggingface-cli download matteogeniaccio/phi-4 --local-dir ./phi-4 --
10
  torch.random.manual_seed(0)
11
 
12
  model = AutoModelForCausalLM.from_pretrained(
13
- "/home/user/app/phi-4/phi-4", # 模型路径
14
  device_map="cuda", # 使用 GPU
15
  torch_dtype="auto", # 自动选择数据类型
16
  trust_remote_code=True, # 允许远程代码加载
17
  )
18
- tokenizer = AutoTokenizer.from_pretrained("/home/user/app/phi-4/phi-4")
19
 
20
  # 设置 pipeline
21
 
 
10
  torch.random.manual_seed(0)
11
 
12
  model = AutoModelForCausalLM.from_pretrained(
13
+ "./phi-4/phi-4", # 模型路径
14
  device_map="cuda", # 使用 GPU
15
  torch_dtype="auto", # 自动选择数据类型
16
  trust_remote_code=True, # 允许远程代码加载
17
  )
18
+ tokenizer = AutoTokenizer.from_pretrained("./phi-4/phi-4")
19
 
20
  # 设置 pipeline
21