Hugging Face's logo

Spaces:
microsoft
/
TRELLIS.2
Running on Zero

App Files Community
6
JeffreyXiang commited on
Commit
691d6b0
·
1 Parent(s): 4f41697

update

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -329,6 +329,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
329
  # Launch the Gradio app
330
  if __name__ == "__main__":
331
  pipeline = Trellis2ImageTo3DPipeline.from_pretrained('JeffreyXiang/TRELLIS.2-4B')
 
332
  pipeline.cuda()
333
 
334
  envmap = EnvMap(torch.tensor(
 
329
  # Launch the Gradio app
330
  if __name__ == "__main__":
331
  pipeline = Trellis2ImageTo3DPipeline.from_pretrained('JeffreyXiang/TRELLIS.2-4B')
332
+ pipeline.low_vram = False
333
  pipeline.cuda()
334
 
335
  envmap = EnvMap(torch.tensor(