Updated requirements to lock version of albucore, which had breaking changes.

2024-09-21 11:19:13 -06:00 · 2024-09-21 11:19:13 -06:00 · d97c55cd96
parent 79b4e04b80
commit d97c55cd96
1 changed files with 2 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -13,7 +13,8 @@ invisible-watermark
 einops
 accelerate
 toml
-albumentations
+albumentations==1.4.15
+albucore==0.0.16
 pydantic
 omegaconf
 k-diffusion