Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
7 |
|
8 |
#Base Model 및 Lora Model 선택
|
9 |
base_model = "EleutherAI/polyglot-ko-5.8b" # "beomi/KoAlpaca-Polyglot-5.8B"
|
10 |
-
lora_weights = "KimSHine/
|
11 |
load_8bit = True
|
12 |
|
13 |
# Base Model Tokenizer
|
|
|
7 |
|
8 |
#Base Model 및 Lora Model 선택
|
9 |
base_model = "EleutherAI/polyglot-ko-5.8b" # "beomi/KoAlpaca-Polyglot-5.8B"
|
10 |
+
lora_weights = "KimSHine/YEOLLM_5.8B-lora_v3" # 'KimSHine/Scenario_Koalpaca_5.8B-lora'
|
11 |
load_8bit = True
|
12 |
|
13 |
# Base Model Tokenizer
|