xinjie.wang commited on
Commit
60a38e8
·
1 Parent(s): 58bf798
Files changed (2) hide show
  1. asset3d_gen/scripts/imageto3d.py +1 -1
  2. 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
- "JeffreyXiang/TRELLIS-image-large"
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
- "JeffreyXiang/TRELLIS-image-large"
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
- "JeffreyXiang/TRELLIS-image-large"
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"