Spaces:
Runtime error
Runtime error
diffuserfix
Browse files- models.py +1 -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.
|
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
|