cavargas10 commited on
Commit
5422690
·
verified ·
1 Parent(s): 3c2a8b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -325,7 +325,7 @@ if __name__ == "__main__":
325
  flux_pipeline = FluxPipeline.from_pretrained(single_file_base_model, transformer=transformer, text_encoder_2=text_encoder_2, torch_dtype=dtype, token=huggingface_token)
326
  flux_pipeline.to("cuda")
327
  # Initialize Trellis pipeline
328
- trellis_pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
329
  trellis_pipeline.cuda()
330
  try:
331
  trellis_pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))
 
325
  flux_pipeline = FluxPipeline.from_pretrained(single_file_base_model, transformer=transformer, text_encoder_2=text_encoder_2, torch_dtype=dtype, token=huggingface_token)
326
  flux_pipeline.to("cuda")
327
  # Initialize Trellis pipeline
328
+ trellis_pipeline = TrellisImageTo3DPipeline.from_pretrained("cavargas10/TRELLIS")
329
  trellis_pipeline.cuda()
330
  try:
331
  trellis_pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))