ldhldh commited on
Commit
6f8b1b3
โ€ข
1 Parent(s): a260c12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ peft_model_id = "ldhldh/1.3_40kstep"
19
  #18k > ์ƒ๋Œ€์˜ ๋ง๊นŒ์ง€ ํ•˜๋Š” ์ด์Šˆ๊ฐ€ ์žˆ์Œ
20
  #8k > ์•ฝ๊ฐ„ ์•„์‰ฌ์šด๊ฐ€?
21
 
22
- base_model = AutoModelForCausalLM.from_pretrained("squarelike/ggml-polyglot-ko-5.8B-q4", hf=True, model_type='gpt_neox')
23
- tokenizer = AutoTokenizer.from_pretrained("EleutherAI/polyglot-ko-5.8b")
24
 
25
  #base_model = AutoModelForCausalLM.from_pretrained("EleutherAI/polyglot-ko-3.8b")
26
  #tokenizer = AutoTokenizer.from_pretrained("EleutherAI/polyglot-ko-3.8b")
 
19
  #18k > ์ƒ๋Œ€์˜ ๋ง๊นŒ์ง€ ํ•˜๋Š” ์ด์Šˆ๊ฐ€ ์žˆ์Œ
20
  #8k > ์•ฝ๊ฐ„ ์•„์‰ฌ์šด๊ฐ€?
21
 
22
+ base_model = AutoModelForCausalLM.from_pretrained("Skyranch/KoAlpaca-Polyglot-12.8B-ggml-model-f16", hf=True, model_type='gpt_neox')
23
+ tokenizer = AutoTokenizer.from_pretrained("beomi/KoAlpaca-Polyglot-12.8B")
24
 
25
  #base_model = AutoModelForCausalLM.from_pretrained("EleutherAI/polyglot-ko-3.8b")
26
  #tokenizer = AutoTokenizer.from_pretrained("EleutherAI/polyglot-ko-3.8b")