refixed

requirements.txt

@@ -16,4 +16,5 @@ jsonmerge
 matplotlib
 resize-right
 timm
-torchdiffeq
+torchdiffeq
+#opencv-python

sd.py

@@ -1,9 +1,9 @@
 import re
-#import cv2
 import pandas as pd
 import numpy as np
 from functools import reduce
 from einops import rearrange
+import sys
 
 from skimage.exposure import match_histograms
 import random
@@ -22,16 +22,17 @@ from omegaconf import OmegaConf
 
 # 1. Download stable diffusion repository and set path
 os.system("git clone https://github.com/deforum/stable-diffusion")
+os.system("git clone https://github.com/deforum/k-diffusion/")
+with open('k-diffusion/k_diffusion/__init__.py', 'w') as f:
+        f.write('')
+
 sys.path.extend([
     'src/taming-transformers',
     'src/clip',
     'stable-diffusion/',
     'k-diffusion',
-    'pytorch3d-lite',
-    'AdaBins',
-    'MiDaS',
 ])
-from helpers import DepthModel, sampler_fn
+from helpers import sampler_fn
 from k_diffusion.external import CompVisDenoiser
 from ldm.util import instantiate_from_config
 from ldm.models.diffusion.ddim import DDIMSampler