ldhldh commited on
Commit
abbd7b9
β€’
1 Parent(s): 9306871

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -11,7 +11,9 @@ 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
  # 20k or 30k
16
  #18k > μƒλŒ€μ˜ λ§κΉŒμ§€ ν•˜λŠ” μ΄μŠˆκ°€ 있음
17
  #8k > μ•½κ°„ μ•„μ‰¬μš΄κ°€?
 
11
  print("Running on device:", torch_device)
12
  print("CPU threads:", torch.get_num_threads())
13
 
14
+ peft_model_id = "ldhldh/1.3_40kstep"
15
+
16
+ #peft_model_id = "ldhldh/polyglot-ko-1.3b_lora_big_tern_30kstep"
17
  # 20k or 30k
18
  #18k > μƒλŒ€μ˜ λ§κΉŒμ§€ ν•˜λŠ” μ΄μŠˆκ°€ 있음
19
  #8k > μ•½κ°„ μ•„μ‰¬μš΄κ°€?