1 files changed, 3 insertions, 3 deletions

diff --git a/environment.yaml b/environment.yaml
index 24693d5..57e090c 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -16,14 +16,14 @@ dependencies:
       - -e .
       - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
       - -e git+https://github.com/openai/CLIP.git@main#egg=clip
-      - -e git+https://github.com/hlky/k-diffusion-sd#egg=k_diffusion
       - -e git+https://github.com/devilismyfriend/latent-diffusion#egg=latent-diffusion
       - -e git+https://github.com/huggingface/diffusers#egg=diffusers
       - accelerate==0.12.0
       - albumentations==1.1.0
-      - bitsandbytes==0.34.0
+      - bitsandbytes==0.35.4
       - einops==0.4.1
       - imageio==2.22.0
+      - k-diffusion==0.0.12
       - kornia==0.6
       - pudb==2019.2
       - omegaconf==2.2.3
@@ -34,5 +34,5 @@ dependencies:
       - test-tube>=0.7.5
       - torch-fidelity==0.3.0
       - torchmetrics==0.9.3
-      - transformers==4.23.1
+      - transformers==4.25.1
       - triton==2.0.0.dev20221105