dwb2023 commited on
Commit
918ac8f
1 Parent(s): 5e9a655

Update app.py

Browse files

add revision info for models.

Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -39,11 +39,12 @@ Based on the [Norod78/OpenELM_3B_Demo](https://huggingface.co/spaces/Norod78/Ope
39
 
40
  model = AutoModelForCausalLM.from_pretrained(
41
  "apple/OpenELM-270M-Instruct",
 
42
  trust_remote_code=True,
43
  )
44
  tokenizer = AutoTokenizer.from_pretrained(
45
- # "NousResearch/Llama-2-7b-hf",
46
- "meta-llama/Llama-2-7b-hf",
47
  trust_remote_code=True,
48
  tokenizer_class=LlamaTokenizer,
49
  )
 
39
 
40
  model = AutoModelForCausalLM.from_pretrained(
41
  "apple/OpenELM-270M-Instruct",
42
+ revision="eb111ff",
43
  trust_remote_code=True,
44
  )
45
  tokenizer = AutoTokenizer.from_pretrained(
46
+ "meta-llama/Llama-2-7b-hf",
47
+ revision="01c7f73"
48
  trust_remote_code=True,
49
  tokenizer_class=LlamaTokenizer,
50
  )