Spaces:
Running on Zero
Running on Zero
Set PYTORCH_CUDA_ALLOC_CONF=expandable_segments to fight fragmentation; explicit low_cpu_mem_usage=False on TAEF1 + add accelerate to requirements to silence the warning
835d190 | accelerate | |
| diffusers>=0.37.0 | |
| transformers==4.57.1 | |
| safetensors | |
| sentencepiece | |
| numpy<2 | |
| pandas | |
| pillow | |
| imageio | |
| opencv-python-headless | |
| einops | |
| hydra-core==1.3.2 | |
| omegaconf==2.3.0 | |
| pyyaml | |
| attrs | |
| attr | |
| loguru | |
| termcolor | |
| fvcore | |
| iopath | |
| pynvml | |
| wandb | |
| ninja | |
| packaging | |
| boto3>=1.34 | |
| botocore>=1.34 | |