diff --git a/requirements.txt b/requirements.txt index 3f3f2c6..b51ef1b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,30 +1,30 @@ -accelerate==0.34.0 -diffusers==0.30.2 -einops==0.7.0 -fla==0.1 -flash-attn==2.5.8 -lightning==2.5.1 +accelerate==1.10.1 +diffusers==0.35.1 +einops==0.8.1 +flash-linear-attention==0.3.2 +flash_attn==2.8.3 +lightning==2.5.5 lpips==0.1.4 -numpy==1.24.4 +numpy==2.1.2 omegaconf==2.3.0 -opencv-python==4.8.0.74 -protobuf==3.19.0 -pytorch-lightning==2.5.1 -PyWavelets==1.6.0 -PyYAML==6.0.1 -safetensors==0.4.3 -scikit-learn==1.2.0 -scipy==1.12.0 -six==1.16.0 -sympy==1.12 -tensorboard==2.9.0 -timm==0.9.11 -tokenizers==0.19.0 +opencv-python==4.12.0.88 +protobuf==6.32.1 +pytorch-lightning==2.5.5 +PyWavelets==1.8.0 +PyYAML==6.0.2 +safetensors==0.6.2 +scikit-learn==1.7.2 +scipy==1.15.3 +six==1.17.0 +sympy==1.13.3 +tensorboard==2.20.0 +timm==1.0.19 +tokenizers==0.22.0 toml==0.10.2 -torch==2.3.0 +torch==2.8.0 torchinfo==1.8.0 -torchmetrics==1.6.0 -torchvision -tqdm==4.66.2 -transformers==4.40.0 -triton==3.2.0 +torchmetrics==1.8.2 +torchvision==0.23.0 +tqdm==4.67.1 +transformers==4.56.1 +triton==3.4.0 \ No newline at end of file