code

app.py

@@ -14,6 +14,7 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="cuda",  # 使用 GPU
     torch_dtype="auto",  # 自动选择数据类型
     trust_remote_code=True,  # 允许远程代码加载
+    use_safetensors=True,
 )
 tokenizer = AutoTokenizer.from_pretrained("./phi-4/phi-4")