jingyangcarl commited on
Commit
f6a3903
·
1 Parent(s): 85dc93e
Files changed (1) hide show
  1. model.py +4 -1
model.py CHANGED
@@ -28,6 +28,7 @@ import subprocess
28
  from rgb2x.pipeline_rgb2x import StableDiffusionAOVMatEstPipeline
29
  from app_texnet import image_to_temp_path
30
  import os
 
31
 
32
  CONTROLNET_MODEL_IDS = {
33
  # "Openpose": "lllyasviel/control_v11p_sd15_openpose",
@@ -317,7 +318,9 @@ class Model:
317
  normal_map_path = image_to_temp_path(preds[0].rotate(90), "normal_map")
318
  roughness_path = image_to_temp_path(preds[1].rotate(90), "roughness")
319
  metallic_path = image_to_temp_path(preds[2].rotate(90), "metallic")
320
- output_blend_path = os.path.join(os.getcwd(), "output", f"{obj_name}.blend") # replace with desired output path
 
 
321
  os.makedirs(os.path.dirname(output_blend_path), exist_ok=True)
322
 
323
  def run_blend_generation(
 
28
  from rgb2x.pipeline_rgb2x import StableDiffusionAOVMatEstPipeline
29
  from app_texnet import image_to_temp_path
30
  import os
31
+ import time
32
 
33
  CONTROLNET_MODEL_IDS = {
34
  # "Openpose": "lllyasviel/control_v11p_sd15_openpose",
 
318
  normal_map_path = image_to_temp_path(preds[0].rotate(90), "normal_map")
319
  roughness_path = image_to_temp_path(preds[1].rotate(90), "roughness")
320
  metallic_path = image_to_temp_path(preds[2].rotate(90), "metallic")
321
+ prompt_nospace = prompt.replace(' ', '_')
322
+ current_timecode = time.strftime("%Y%m%d_%H%M%S")
323
+ output_blend_path = os.path.join(os.getcwd(), "output", f"{obj_name}_{prompt_nospace}_{current_timecode}.blend") # replace with desired output path
324
  os.makedirs(os.path.dirname(output_blend_path), exist_ok=True)
325
 
326
  def run_blend_generation(