Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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("
|
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)))
|