openfree commited on
Commit
071e81f
·
verified ·
1 Parent(s): aada850

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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
- config["config"]["process"][0]["model"]["name_or_path"] = "black-forest-labs/FLUX.1-schnell"
225
- config["config"]["process"][0]["model"]["assistant_lora_path"] = "ostris/FLUX.1-schnell-training-adapter"
226
- config["config"]["process"][0]["sample"]["sample_steps"] = 4
 
 
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)