xuxw98 commited on
Commit
457689b
1 Parent(s): d0abca8

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -109,9 +109,9 @@ def instruct_generate(
109
  pretrained_path = hf_hub_download(
110
  repo_id="Gary3410/pretrain_lit_llama", filename="lit-llama.pth")
111
  tokenizer_path = hf_hub_download(
112
- repo_id="Gary3410/tapa_model", filename="tokenizer.model")
113
  adapter_path = hf_hub_download(
114
- repo_id="Gary3410/tapa_model", filename="lit-llama-adapter-finetuned_15k.pth")
115
  # adapter_path = "lit-llama-adapter-finetuned_15k.pth"
116
  example_path = "example.json"
117
  # 1024如果不够, 调整为512
 
109
  pretrained_path = hf_hub_download(
110
  repo_id="Gary3410/pretrain_lit_llama", filename="lit-llama.pth")
111
  tokenizer_path = hf_hub_download(
112
+ repo_id="Gary3410/pretrain_lit_llama", filename="tokenizer.model")
113
  adapter_path = hf_hub_download(
114
+ repo_id="Gary3410/pretrain_lit_llama", filename="lit-llama-adapter-finetuned_15k.pth")
115
  # adapter_path = "lit-llama-adapter-finetuned_15k.pth"
116
  example_path = "example.json"
117
  # 1024如果不够, 调整为512