Spaces:
Build error
Build 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
|