syedMohib44 commited on
Commit
68df72b
·
1 Parent(s): fefdc9f

update model path

Browse files
app.py CHANGED
@@ -148,7 +148,7 @@ class ModelWorker:
148
  logger.info(f"Loading the model {model_path} on worker {worker_id} ...")
149
 
150
  self.rembg = BackgroundRemover()
151
- self.pipeline = Hunyuan3DDiTFlowMatchingPipeline.from_pretrained(model_path, device=device)
152
  # self.pipeline_t2i = HunyuanDiTPipeline('Tencent-Hunyuan/HunyuanDiT-v1.1-Diffusers-Distilled',
153
  # device=device)
154
  self.pipeline_tex = Hunyuan3DPaintPipeline.from_pretrained(model_path)
 
148
  logger.info(f"Loading the model {model_path} on worker {worker_id} ...")
149
 
150
  self.rembg = BackgroundRemover()
151
+ self.pipeline = Hunyuan3DDiTFlowMatchingPipeline.from_pretrained(model_path, cache_dir='content/ditto-api/tencent/Hunyuan3D-2', device=device)
152
  # self.pipeline_t2i = HunyuanDiTPipeline('Tencent-Hunyuan/HunyuanDiT-v1.1-Diffusers-Distilled',
153
  # device=device)
154
  self.pipeline_tex = Hunyuan3DPaintPipeline.from_pretrained(model_path)
hy3dgen/shapegen/pipelines.py CHANGED
@@ -210,7 +210,7 @@ class Hunyuan3DDiTPipeline:
210
  original_model_path = model_path
211
  if not os.path.exists(model_path):
212
  # try local path
213
- base_dir = os.environ.get('HY3DGEN_MODELS', '/content/hy3dgen')
214
  model_path = os.path.expanduser(os.path.join(base_dir, model_path, subfolder))
215
  if not os.path.exists(model_path):
216
  try:
 
210
  original_model_path = model_path
211
  if not os.path.exists(model_path):
212
  # try local path
213
+ base_dir = os.environ.get('HY3DGEN_MODELS', '/content/ditto-api/tencent/Hunyuan3D-2/hy3dgen')
214
  model_path = os.path.expanduser(os.path.join(base_dir, model_path, subfolder))
215
  if not os.path.exists(model_path):
216
  try:
hy3dgen/texgen/pipelines.py CHANGED
@@ -61,7 +61,7 @@ class Hunyuan3DPaintPipeline:
61
  original_model_path = model_path
62
  if not os.path.exists(model_path):
63
  # try local path
64
- base_dir = os.environ.get('HY3DGEN_MODELS', '/content/hy3dgen')
65
  model_path = os.path.expanduser(os.path.join(base_dir, model_path))
66
 
67
  delight_model_path = os.path.join(model_path, 'hunyuan3d-delight-v2-0')
 
61
  original_model_path = model_path
62
  if not os.path.exists(model_path):
63
  # try local path
64
+ base_dir = os.environ.get('HY3DGEN_MODELS', '//content/ditto-api/tencent/Hunyuan3D-2/hy3dgen')
65
  model_path = os.path.expanduser(os.path.join(base_dir, model_path))
66
 
67
  delight_model_path = os.path.join(model_path, 'hunyuan3d-delight-v2-0')