jacob-c commited on
Commit
12e6745
·
1 Parent(s): 80cfe38

diffuserfix

Browse files
Files changed (2) hide show
  1. models.py +1 -2
  2. requirements.txt +1 -1
models.py CHANGED
@@ -4,7 +4,6 @@ import random
4
  import inspect
5
  import numpy as np
6
  from tqdm import tqdm
7
- import torch
8
 
9
  import torch
10
  import torch.nn as nn
@@ -24,7 +23,7 @@ import sys
24
  sys.path.insert(0, "diffusers/src")
25
 
26
  import diffusers
27
- from diffusers.utils import randn_tensor
28
  from diffusers import DDPMScheduler, UNet2DConditionModel
29
  from diffusers import AutoencoderKL as DiffuserAutoencoderKL
30
 
 
4
  import inspect
5
  import numpy as np
6
  from tqdm import tqdm
 
7
 
8
  import torch
9
  import torch.nn as nn
 
23
  sys.path.insert(0, "diffusers/src")
24
 
25
  import diffusers
26
+ from diffusers.utils.torch_utils import randn_tensor
27
  from diffusers import DDPMScheduler, UNet2DConditionModel
28
  from diffusers import AutoencoderKL as DiffuserAutoencoderKL
29
 
requirements.txt CHANGED
@@ -7,7 +7,7 @@ datasets==2.1.0
7
  einops==0.6.1
8
  h5py==3.8.0
9
  huggingface_hub>=0.23.4
10
- diffusers==0.10.0
11
  importlib_metadata==6.3.0
12
  librosa==0.9.2
13
  matplotlib==3.5.2
 
7
  einops==0.6.1
8
  h5py==3.8.0
9
  huggingface_hub>=0.23.4
10
+ diffusers==0.21.4
11
  importlib_metadata==6.3.0
12
  librosa==0.9.2
13
  matplotlib==3.5.2