JUNGU commited on
Commit
13bee58
โ€ข
1 Parent(s): fafb726

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -24,14 +24,15 @@ token = os.environ["HF_TOKEN"]
24
  # apple/OpenELM-270M ๋ชจ๋ธ๊ณผ ํ† ํฌ๋‚˜์ด์ € ๋กœ๋“œ
25
  # ํ† ํฌ๋‚˜์ด์ €๊ฐ€ ์˜ค๋ฅ˜๋‚˜๋Š” ๋ฌธ์ œ๊ฐ€ ์žˆ์–ด์„œ NousResearch/Llama-2-7b-hf๋ฅผ ์”€
26
  # ํ•œ๊ตญ์–ด ๋ชจ๋ธ ํ† ํฌ๋‚˜์ด์ €๋กœ ๋ฐ”๊ฟ”๋ด„ beomi/llama-2-ko-7b
27
- # apple/OpenELM-1.1B ํ† ํฌ๋‚˜์ด์ €๋งŒ ํฌ๊ฒŒ ํ•ด๋ด„๋ด„
 
28
  model = AutoModelForCausalLM.from_pretrained(
29
  "apple/OpenELM-270M",
30
  token=token,
31
  trust_remote_code=True,
32
  )
33
  tok = AutoTokenizer.from_pretrained(
34
- "apple/OpenELM-1_1B",
35
  token=token,
36
  trust_remote_code=True,
37
  tokenizer_class=LlamaTokenizer,
 
24
  # apple/OpenELM-270M ๋ชจ๋ธ๊ณผ ํ† ํฌ๋‚˜์ด์ € ๋กœ๋“œ
25
  # ํ† ํฌ๋‚˜์ด์ €๊ฐ€ ์˜ค๋ฅ˜๋‚˜๋Š” ๋ฌธ์ œ๊ฐ€ ์žˆ์–ด์„œ NousResearch/Llama-2-7b-hf๋ฅผ ์”€
26
  # ํ•œ๊ตญ์–ด ๋ชจ๋ธ ํ† ํฌ๋‚˜์ด์ €๋กœ ๋ฐ”๊ฟ”๋ด„ beomi/llama-2-ko-7b
27
+ # apple/OpenELM-1.1B ํ† ํฌ๋‚˜์ด์ €๋งŒ ํฌ๊ฒŒ ํ•ด๋ด„ <- ์•ˆ๋จ
28
+ #
29
  model = AutoModelForCausalLM.from_pretrained(
30
  "apple/OpenELM-270M",
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,