Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
•
4649643
1
Parent(s):
2740e75
Update app.py
Browse files
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 |
-
|
|
|
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)
|