Shangding-Gu commited on
Commit
a0b0102
·
1 Parent(s): 0b8b900

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ import json
11
 
12
  assert (
13
  "LlamaTokenizer" in transformers._import_structure["models.llama"]
14
- ), "LLaMA is now in HuggingFace's main branch.\nPlease reinstall it: pip uninstall transformers && pip install git+https://github.com/huggingface/transformers.git"
15
  from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
16
 
17
  if torch.cuda.is_available():
@@ -26,7 +26,7 @@ except:
26
  pass
27
 
28
 
29
- base_model = "/home/v-shanggu/MyCode_server_1503/Llama-X/RL_evol_llama2_chat_7B/checkpoint-40" # "/path/to/WizardLM13B",
30
 
31
  tokenizer = LlamaTokenizer.from_pretrained(base_model)
32
  load_8bit = False
 
11
 
12
  assert (
13
  "LlamaTokenizer" in transformers._import_structure["models.llama"]
14
+ ), "Please reinstall it: pip uninstall transformers && pip install git+https://github.com/huggingface/transformers.git"
15
  from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
16
 
17
  if torch.cuda.is_available():
 
26
  pass
27
 
28
 
29
+ base_model = "https://huggingface.co/Shangding-Gu/Lunyu-LLM/"
30
 
31
  tokenizer = LlamaTokenizer.from_pretrained(base_model)
32
  load_8bit = False