jason9693 commited on
Commit
dde792a
·
1 Parent(s): 621210f

model to 2.7b

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,9 +15,9 @@ examples = [
15
  ["2040년 미국은, "]
16
  ]
17
 
18
- tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neox-ko-1.3b", use_auth_token=os.environ['TOKEN'])
19
  model = AutoModelForCausalLM.from_pretrained(
20
- "EleutherAI/gpt-neox-ko-1.3b", use_auth_token=os.environ['TOKEN'],
21
  pad_token_id=tokenizer.pad_token_id,
22
  eos_token_id=tokenizer.eos_token_id,
23
  bos_token_id=tokenizer.bos_token_id
 
15
  ["2040년 미국은, "]
16
  ]
17
 
18
+ tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neox-ko-2.7b", use_auth_token=os.environ['TOKEN'])
19
  model = AutoModelForCausalLM.from_pretrained(
20
+ "EleutherAI/gpt-neox-ko-2.7b", use_auth_token=os.environ['TOKEN'],
21
  pad_token_id=tokenizer.pad_token_id,
22
  eos_token_id=tokenizer.eos_token_id,
23
  bos_token_id=tokenizer.bos_token_id