code

app.py

@@ -4,7 +4,8 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import os
 
-os.system('huggingface-cli download matteogeniaccio/phi-4 --local-dir ./phi-4 --include "phi-4/*"')
+if not os.path.exists("./phi-4"):
+    os.system('huggingface-cli download matteogeniaccio/phi-4 --local-dir ./phi-4 --include "phi-4/*"')
 
 from safetensors.torch import load_file, save_file
 
@@ -30,7 +31,8 @@ def merge_safetensors(input_dir, output_file):
 input_dir = "./phi-4/phi-4"  # 包含分片文件的目录
 output_file = "./phi-4/phi-4/model.safetensors"  # 合并后的文件路径
 
-merge_safetensors(input_dir, output_file)
+if not os.path.exists(output_file):
+    merge_safetensors(input_dir, output_file)
 
 # 加载 phi-4 模型和 tokenizer
 torch.random.manual_seed(0)