JUNGU commited on
Commit
ecf6383
โ€ข
1 Parent(s): 5791887

Update app.py

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