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)