JUNGU commited on
Commit
5791887
1 Parent(s): 2f7b0c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,12 +27,12 @@ token = os.environ["HF_TOKEN"]
27
  # apple/OpenELM-1.1B 토크나이저만 크게 해봄 <- 안됨
28
  # OpenELM-3B-Instruct로 둘다 변경경
29
  model = AutoModelForCausalLM.from_pretrained(
30
- "OpenELM-3B-Instruct",
31
  token=token,
32
  trust_remote_code=True,
33
  )
34
  tok = AutoTokenizer.from_pretrained(
35
- "NousResearch/Llama-2-7b-hf",
36
  token=token,
37
  trust_remote_code=True,
38
  tokenizer_class=LlamaTokenizer,
 
27
  # apple/OpenELM-1.1B 토크나이저만 크게 해봄 <- 안됨
28
  # OpenELM-3B-Instruct로 둘다 변경경
29
  model = AutoModelForCausalLM.from_pretrained(
30
+ "apple/OpenELM-3B-Instruct",
31
  token=token,
32
  trust_remote_code=True,
33
  )
34
  tok = AutoTokenizer.from_pretrained(
35
+ "apple/OpenELM-3B-Instruct",
36
  token=token,
37
  trust_remote_code=True,
38
  tokenizer_class=LlamaTokenizer,