Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -221,9 +221,11 @@ def start_training(
|
|
221 |
config["config"]["process"][0]["train"]["disable_sampling"] = True
|
222 |
|
223 |
if(which_model == "[schnell] (4 step fast model)"):
|
224 |
-
|
225 |
-
config["config"]["process"][0]["model"]["
|
226 |
-
config["config"]["process"][0]["
|
|
|
|
|
227 |
|
228 |
if(use_more_advanced_options):
|
229 |
more_advanced_options_dict = yaml.safe_load(more_advanced_options)
|
|
|
221 |
config["config"]["process"][0]["train"]["disable_sampling"] = True
|
222 |
|
223 |
if(which_model == "[schnell] (4 step fast model)"):
|
224 |
+
# schnell 관련 조건문을 dev로 변경
|
225 |
+
config["config"]["process"][0]["model"]["name_or_path"] = "black-forest-labs/FLUX.1-dev"
|
226 |
+
config["config"]["process"][0]["model"]["assistant_lora_path"] = "ostris/FLUX.1-dev-training-adapter"
|
227 |
+
config["config"]["process"][0]["sample"]["sample_steps"] = 28 # dev 모델의 기본 스텝
|
228 |
+
|
229 |
|
230 |
if(use_more_advanced_options):
|
231 |
more_advanced_options_dict = yaml.safe_load(more_advanced_options)
|