Spaces:

vibs08
/

image-3d-fastapi

Runtime error

vibs08 commited on Aug 27, 2024

Commit

61308b0

verified ·

1 Parent(s): 4be8881

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ model = TSR.from_pretrained(
     config_name="config.yaml",
     weight_name="model.ckpt",
 )
-model.renderer.set_chunk_size(131072)
 model.to(device)
 rembg_session = rembg.new_session()

     config_name="config.yaml",
     weight_name="model.ckpt",
 )
+model.renderer.set_chunk_size(104218)
 model.to(device)
 rembg_session = rembg.new_session()