ldhldh commited on
Commit
a1a37d8
ยท
1 Parent(s): c0fd6c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ torch_device = "cuda" if torch.cuda.is_available() else "cpu"
11
  print("Running on device:", torch_device)
12
  print("CPU threads:", torch.get_num_threads())
13
 
14
- peft_model_id = "ldhldh/polyglot-ko-1.3b_lora_big_tern_30kstep"
 
15
  #18k > ์ƒ๋Œ€์˜ ๋ง๊นŒ์ง€ ํ•˜๋Š” ์ด์Šˆ๊ฐ€ ์žˆ์Œ
16
  #8k > ์•ฝ๊ฐ„ ์•„์‰ฌ์šด๊ฐ€?
17
  config = PeftConfig.from_pretrained(peft_model_id)
 
11
  print("Running on device:", torch_device)
12
  print("CPU threads:", torch.get_num_threads())
13
 
14
+ peft_model_id = "ldhldh/polyglot-ko-1.3b_lora_big_tern_20kstep"
15
+ # 20k or 30k
16
  #18k > ์ƒ๋Œ€์˜ ๋ง๊นŒ์ง€ ํ•˜๋Š” ์ด์Šˆ๊ฐ€ ์žˆ์Œ
17
  #8k > ์•ฝ๊ฐ„ ์•„์‰ฌ์šด๊ฐ€?
18
  config = PeftConfig.from_pretrained(peft_model_id)