DeepBeepMeep commited on
Commit ·
33a42d1
1
Parent(s): 98476f8
Save generation parameters in a file
Browse files- gradio_server.py +15 -2
gradio_server.py
CHANGED
|
@@ -1237,6 +1237,19 @@ def generate_video(
|
|
| 1237 |
nrow=1,
|
| 1238 |
normalize=True,
|
| 1239 |
value_range=(-1, 1))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1240 |
|
| 1241 |
print(f"New video saved to Path: "+video_path)
|
| 1242 |
file_list.append(video_path)
|
|
@@ -1306,7 +1319,7 @@ def save_lset(state, lset_name, loras_choices, loras_mult_choices, prompt, save_
|
|
| 1306 |
full_lset_name_filename = os.path.join(lora_dir, lset_name_filename)
|
| 1307 |
|
| 1308 |
with open(full_lset_name_filename, "w", encoding="utf-8") as writer:
|
| 1309 |
-
writer.write(json.dumps(lset))
|
| 1310 |
|
| 1311 |
if lset_name in loras_presets:
|
| 1312 |
gr.Info(f"Lora Preset '{lset_name}' has been updated")
|
|
@@ -1582,7 +1595,7 @@ def create_demo():
|
|
| 1582 |
}
|
| 1583 |
"""
|
| 1584 |
default_flow_shift = get_default_flow(transformer_filename_i2v if use_image2video else transformer_filename_t2v)
|
| 1585 |
-
with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="
|
| 1586 |
state_dict = {}
|
| 1587 |
|
| 1588 |
if use_image2video:
|
|
|
|
| 1237 |
nrow=1,
|
| 1238 |
normalize=True,
|
| 1239 |
value_range=(-1, 1))
|
| 1240 |
+
|
| 1241 |
+
configs = {
|
| 1242 |
+
'prompt': prompt,
|
| 1243 |
+
'negative_prompt': negative_prompt,
|
| 1244 |
+
'resolution': resolution,
|
| 1245 |
+
'video_length': video_length,
|
| 1246 |
+
'seed': seed,
|
| 1247 |
+
'num_inference_steps': num_inference_steps,
|
| 1248 |
+
}
|
| 1249 |
+
|
| 1250 |
+
# Save the configs as json format
|
| 1251 |
+
with open(video_path.replace('.mp4', '.json'), 'w') as f:
|
| 1252 |
+
json.dump(configs, f, indent=4)
|
| 1253 |
|
| 1254 |
print(f"New video saved to Path: "+video_path)
|
| 1255 |
file_list.append(video_path)
|
|
|
|
| 1319 |
full_lset_name_filename = os.path.join(lora_dir, lset_name_filename)
|
| 1320 |
|
| 1321 |
with open(full_lset_name_filename, "w", encoding="utf-8") as writer:
|
| 1322 |
+
writer.write(json.dumps(lset, indent=4))
|
| 1323 |
|
| 1324 |
if lset_name in loras_presets:
|
| 1325 |
gr.Info(f"Lora Preset '{lset_name}' has been updated")
|
|
|
|
| 1595 |
}
|
| 1596 |
"""
|
| 1597 |
default_flow_shift = get_default_flow(transformer_filename_i2v if use_image2video else transformer_filename_t2v)
|
| 1598 |
+
with gr.Blocks(css=css, theme=gr.themes.Soft(primary_hue="sky", neutral_hue="slate", text_size= "md")) as demo:
|
| 1599 |
state_dict = {}
|
| 1600 |
|
| 1601 |
if use_image2video:
|