Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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)
|