Update no_need_requirements.txt

2024-08-06 15:31:23 +08:00
parent 2cd7ca71c3
commit dfc2bfb602
1 changed files with 21 additions and 1 deletions
@@ -15,4 +15,24 @@ omegaconf==2.1.1
 -e git+https://github.com/crowsonkb/k-diffusion.git@v0.0.16#egg=k-diffusion
 -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
 imageio==2.9.0
-imageio-ffmpeg==0.4.2
+imageio-ffmpeg==0.4.2
+
+
+# K_diffusion need 
+accelerate
+clean-fid
+clip-anytorch
+dctorch
+einops
+jsonmerge
+kornia
+Pillow
+safetensors
+scikit-image
+scipy
+torch>=2.1
+torchdiffeq
+torchsde
+torchvision
+tqdm
+wandb