airplane194 commited on
Commit
4f00e0c
·
1 Parent(s): a643d17
Files changed (1) hide show
  1. python.py +2 -1
python.py CHANGED
@@ -11,6 +11,7 @@ from custom_nodes.ComfyUI_Comfyroll_CustomNodes.node_mappings import NODE_CLASS_
11
  from custom_nodes.ComfyUI_Comfyroll_CustomNodes.node_mappings import NODE_CLASS_MAPPINGS as NODE_CLASS_MAPPINGS_4
12
  from comfy_extras.nodes_model_advanced import NODE_CLASS_MAPPINGS as NODE_CLASS_MAPPINGS_5
13
  from comfy_extras.nodes_flux import NODE_CLASS_MAPPINGS as NODE_CLASS_MAPPINGS_6
 
14
 
15
  from huggingface_hub import hf_hub_download
16
  token = os.getenv("HUGGINGFACE_TOKEN")
@@ -265,5 +266,5 @@ def generate_image(prompt,
265
  saveimage_9 = saveimage.save_images(
266
  filename_prefix="image", images=get_value_at_index(vaedecode_8, 0)
267
  )
268
- saved_path = f"output/{saveimage_9['ui']['images'][0]['filename']}"
269
  return saved_path, seed
 
11
  from custom_nodes.ComfyUI_Comfyroll_CustomNodes.node_mappings import NODE_CLASS_MAPPINGS as NODE_CLASS_MAPPINGS_4
12
  from comfy_extras.nodes_model_advanced import NODE_CLASS_MAPPINGS as NODE_CLASS_MAPPINGS_5
13
  from comfy_extras.nodes_flux import NODE_CLASS_MAPPINGS as NODE_CLASS_MAPPINGS_6
14
+ import time
15
 
16
  from huggingface_hub import hf_hub_download
17
  token = os.getenv("HUGGINGFACE_TOKEN")
 
266
  saveimage_9 = saveimage.save_images(
267
  filename_prefix="image", images=get_value_at_index(vaedecode_8, 0)
268
  )
269
+ saved_path = f"output/image_{int(time.time() * 1000)}_{saveimage_9['ui']['images'][0]['filename']}"
270
  return saved_path, seed