Spaces:
Running
on
Zero
Running
on
Zero
xinjie.wang
commited on
Commit
·
60a38e8
1
Parent(s):
58bf798
update
Browse files- asset3d_gen/scripts/imageto3d.py +1 -1
- common.py +2 -2
asset3d_gen/scripts/imageto3d.py
CHANGED
@@ -74,7 +74,7 @@ RBG_REMOVER = RembgRemover()
|
|
74 |
RBG14_REMOVER = BMGG14Remover()
|
75 |
SAM_PREDICTOR = SAMPredictor(model_type="vit_h", device="cpu")
|
76 |
PIPELINE = TrellisImageTo3DPipeline.from_pretrained(
|
77 |
-
"
|
78 |
)
|
79 |
PIPELINE.cuda()
|
80 |
SEG_CHECKER = ImageSegChecker(GPT_CLIENT)
|
|
|
74 |
RBG14_REMOVER = BMGG14Remover()
|
75 |
SAM_PREDICTOR = SAMPredictor(model_type="vit_h", device="cpu")
|
76 |
PIPELINE = TrellisImageTo3DPipeline.from_pretrained(
|
77 |
+
"microsoft/TRELLIS-image-large"
|
78 |
)
|
79 |
PIPELINE.cuda()
|
80 |
SEG_CHECKER = ImageSegChecker(GPT_CLIENT)
|
common.py
CHANGED
@@ -157,7 +157,7 @@ if os.getenv("GRADIO_APP") == "imageto3d":
|
|
157 |
RBG14_REMOVER = BMGG14Remover()
|
158 |
SAM_PREDICTOR = SAMPredictor(model_type="vit_h", device="cpu")
|
159 |
PIPELINE = TrellisImageTo3DPipeline.from_pretrained(
|
160 |
-
"
|
161 |
)
|
162 |
# PIPELINE.cuda()
|
163 |
SEG_CHECKER = ImageSegChecker(GPT_CLIENT)
|
@@ -171,7 +171,7 @@ elif os.getenv("GRADIO_APP") == "textto3d":
|
|
171 |
RBG_REMOVER = RembgRemover()
|
172 |
RBG14_REMOVER = BMGG14Remover()
|
173 |
PIPELINE = TrellisImageTo3DPipeline.from_pretrained(
|
174 |
-
"
|
175 |
)
|
176 |
# PIPELINE.cuda()
|
177 |
text_model_dir = "weights/Kolors"
|
|
|
157 |
RBG14_REMOVER = BMGG14Remover()
|
158 |
SAM_PREDICTOR = SAMPredictor(model_type="vit_h", device="cpu")
|
159 |
PIPELINE = TrellisImageTo3DPipeline.from_pretrained(
|
160 |
+
"microsoft/TRELLIS-image-large"
|
161 |
)
|
162 |
# PIPELINE.cuda()
|
163 |
SEG_CHECKER = ImageSegChecker(GPT_CLIENT)
|
|
|
171 |
RBG_REMOVER = RembgRemover()
|
172 |
RBG14_REMOVER = BMGG14Remover()
|
173 |
PIPELINE = TrellisImageTo3DPipeline.from_pretrained(
|
174 |
+
"microsoft/TRELLIS-image-large"
|
175 |
)
|
176 |
# PIPELINE.cuda()
|
177 |
text_model_dir = "weights/Kolors"
|