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 > ์•ฝ๊ฐ„ ์•„์‰ฌ์šด๊ฐ€?