frogleo commited on
Commit
6d4f78d
·
verified ·
1 Parent(s): a65c25e

Update hy3dgen/texgen/pipelines.py

Browse files
Files changed (1) hide show
  1. hy3dgen/texgen/pipelines.py +3 -2
hy3dgen/texgen/pipelines.py CHANGED
@@ -222,8 +222,9 @@ class Hunyuan3DPaintPipeline:
222
  image_prompt = image
223
 
224
  image_prompt = self.recenter_image(image_prompt)
225
-
226
- image_prompt = self.models['delight_model'](image_prompt)
 
227
 
228
  mesh = mesh_uv_wrap(mesh)
229
 
 
222
  image_prompt = image
223
 
224
  image_prompt = self.recenter_image(image_prompt)
225
+
226
+ if self.config.use_delight:
227
+ image_prompt = self.models['delight_model'](image_prompt)
228
 
229
  mesh = mesh_uv_wrap(mesh)
230