Spaces:
Runtime error
Runtime error
syedMohib44
commited on
Commit
·
68df72b
1
Parent(s):
fefdc9f
update model path
Browse files- app.py +1 -1
- hy3dgen/shapegen/pipelines.py +1 -1
- hy3dgen/texgen/pipelines.py +1 -1
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', '
|
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')
|