FramePack can be trained with a much larger batch size, similar to the batch size for image diffusion training. Nvidia GPU in RTX 30XX, 40XX, 50XX series that supports fp16 and bf16. The GTX 10XX/20XX ...

cd .. pip install kaleido plotly open3d tyro termcolor h5py cd .. # install 3d diffusion policy pip install --no-cache-dir wandb ipdb gpustat visdom notebook mediapy torch_geometric natsort ...