Remove dependency version constraints (resolves #1)
This commit is contained in:
@@ -20,7 +20,7 @@ from torch import nn
|
||||
from diffusers.utils import is_torch_version
|
||||
from diffusers.utils.torch_utils import apply_freeu
|
||||
from diffusers.models.attention import Attention
|
||||
from diffusers.models.dual_transformer_2d import DualTransformer2DModel
|
||||
from diffusers.models.transformers.dual_transformer_2d import DualTransformer2DModel
|
||||
from diffusers.models.resnet import (
|
||||
Downsample2D,
|
||||
ResnetBlock2D,
|
||||
@@ -28,7 +28,7 @@ from diffusers.models.resnet import (
|
||||
TemporalConvLayer,
|
||||
Upsample2D,
|
||||
)
|
||||
from diffusers.models.transformer_2d import Transformer2DModel
|
||||
from diffusers.models.transformers.transformer_2d import Transformer2DModel
|
||||
from .transformer_temporal import (
|
||||
TransformerSpatioTemporalModel,
|
||||
TransformerTemporalModel,
|
||||
|
||||
Reference in New Issue
Block a user