Hugging Face's logo

Spaces:
rayli
/
DragAPart
Running on Zero

App Files Community
3
rayli commited on
Commit
038e420
•
1 Parent(s): 30b64bc

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -40,8 +40,8 @@ from diffusers.models.embeddings import (
40
  from diffusers.models.activations import get_activation
41
  from diffusers.models.attention_processor import AttentionProcessor, AttnProcessor, Attention, AttnAddedKVProcessor, AttnAddedKVProcessor2_0
42
  from diffusers.models.resnet import Downsample2D, FirDownsample2D, FirUpsample2D, KDownsample2D, KUpsample2D, ResnetBlock2D, Upsample2D
43
- from diffusers.models.transformer_2d import Transformer2DModel
44
- from diffusers.models.dual_transformer_2d import DualTransformer2DModel
45
 
46
 
47
  class CrossAttnDownBlock2DWithFlow(nn.Module):
 
40
  from diffusers.models.activations import get_activation
41
  from diffusers.models.attention_processor import AttentionProcessor, AttnProcessor, Attention, AttnAddedKVProcessor, AttnAddedKVProcessor2_0
42
  from diffusers.models.resnet import Downsample2D, FirDownsample2D, FirUpsample2D, KDownsample2D, KUpsample2D, ResnetBlock2D, Upsample2D
43
+ from diffusers.models.transformers.transformer_2d import Transformer2DModel
44
+ from diffusers.models.transformers.dual_transformer_2d import DualTransformer2DModel
45
 
46
 
47
  class CrossAttnDownBlock2DWithFlow(nn.Module):