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', '/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')
|