DeepBeepMeep commited on
Commit
33a42d1
·
1 Parent(s): 98476f8

Save generation parameters in a file

Browse files
Files changed (1) hide show
  1. 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="emerald", neutral_hue="slate", text_size= "md")) as demo:
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: