update some packages

app.py

@@ -25,6 +25,7 @@ from src.utils.infer_util import (remove_background, resize_foreground)
 from src.utils.mesh_util import save_glb, save_obj
 from src.utils.train_util import instantiate_from_config
 
+torch.backends.cuda.matmul.allow_tf32 = True
 
 def find_cuda():
     cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')
@@ -286,4 +287,4 @@ with gr.Blocks() as demo:
         outputs=[output_model_obj, output_model_glb]
     )
 
-demo.launch()
+demo.launch()

requirements.txt

@@ -1,7 +1,7 @@
-torch==2.1.0
-torchvision==0.16.0
-torchaudio==2.1.0
-pytorch-lightning==2.1.2
+torch==2.3.0
+torchvision==0.18.0
+torchaudio==2.3.0
+pytorch-lightning==2.2.4
 einops
 omegaconf
 deepspeed
@@ -12,8 +12,8 @@ tensorboard
 PyMCubes
 trimesh
 rembg
-transformers==4.38.2
-diffusers==0.25.0
+transformers==4.40.2
+diffusers==0.27.2
 bitsandbytes
 imageio[ffmpeg]
 xatlas