multimodalart HF staff commited on
Commit
4649643
1 Parent(s): 2740e75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -213,9 +213,11 @@ def start_training(
213
  config["config"]["process"][0]["save"]["hf_private"] = True
214
  if concept_sentence:
215
  config["config"]["process"][0]["trigger_word"] = concept_sentence
216
- if sample_1 or sample_2 or sample_2:
 
217
  config["config"]["process"][0]["train"]["disable_sampling"] = False
218
  config["config"]["process"][0]["sample"]["sample_every"] = steps
 
219
  config["config"]["process"][0]["sample"]["prompts"] = []
220
  if sample_1:
221
  config["config"]["process"][0]["sample"]["prompts"].append(sample_1)
 
213
  config["config"]["process"][0]["save"]["hf_private"] = True
214
  if concept_sentence:
215
  config["config"]["process"][0]["trigger_word"] = concept_sentence
216
+
217
+ if sample_1 or sample_2 or sample_3:
218
  config["config"]["process"][0]["train"]["disable_sampling"] = False
219
  config["config"]["process"][0]["sample"]["sample_every"] = steps
220
+ config["config"]["process"][0]["sample"]["sample_steps"] = 28
221
  config["config"]["process"][0]["sample"]["prompts"] = []
222
  if sample_1:
223
  config["config"]["process"][0]["sample"]["prompts"].append(sample_1)