Upload app.py

app.py

@@ -108,13 +108,14 @@ def instruct_generate(
     return output
 
 # 配置具体参数
-pretrained_path = hf_hub_download(
-    repo_id="Gary3410/pretrain_lit_llama", filename="lit-llama.pth")
+# pretrained_path = hf_hub_download(
+#     repo_id="Gary3410/pretrain_lit_llama", filename="lit-llama.pth")
 tokenizer_path = hf_hub_download(
     repo_id="Gary3410/pretrain_lit_llama", filename="tokenizer.model")
 adapter_path = hf_hub_download(
     repo_id="Gary3410/pretrain_lit_llama", filename="lit-llama-adapter-finetuned_15k.pth")
 # adapter_path = "lit-llama-adapter-finetuned_15k.pth"
+pretrained_path = "lit-llama.pth"
 example_path = "example.json"
 # 1024如果不够, 调整为512
 max_seq_len = 1024