Spaces:

likewendy
/

phi-4

Sleeping

likewendy commited on Jan 6

Commit

8e50b08

1 Parent(s): 859e94e

code

Files changed (1) hide show

app.py CHANGED Viewed

@@ -40,7 +40,7 @@ torch.random.manual_seed(0)
 model = AutoModelForCausalLM.from_pretrained(
     "./phi-4/phi-4",  # 模型路径
     device_map="cuda",  # 使用 GPU
-    torch_dtype="auto",  # 自动选择数据类型
     trust_remote_code=True,  # 允许远程代码加载
     use_safetensors=True,
     torch_dtype=torch.float16,

 model = AutoModelForCausalLM.from_pretrained(
     "./phi-4/phi-4",  # 模型路径
     device_map="cuda",  # 使用 GPU
+    # torch_dtype="auto",  # 自动选择数据类型
     trust_remote_code=True,  # 允许远程代码加载
     use_safetensors=True,
     torch_dtype=torch.float16,