fix diffusers

genphoto/models/attention.py

@@ -6,10 +6,10 @@ from typing import Optional
 import torch
 from torch import nn
 
-from diffusers.configuration_utils import ConfigMixin, register_to_config
-from diffusers.models.modeling_utils import ModelMixin
-from diffusers.utils import BaseOutput
-from diffusers.models.attention import BasicTransformerBlock
+from ddiffusers.configuration_utils import ConfigMixin, register_to_config
+from ddiffusers.models.modeling_utils import ModelMixin
+from ddiffusers.utils import BaseOutput
+from ddiffusers.models.attention import BasicTransformerBlock
 from einops import rearrange, repeat
 
 

genphoto/models/attention_processor.py

@@ -2,9 +2,9 @@ import torch
 import torch.nn as nn
 import torch.nn.init as init
 import logging
-from diffusers.models.lora import LoRALinearLayer
-from diffusers.models.attention import Attention
-from diffusers.utils import USE_PEFT_BACKEND
+from ddiffusers.models.lora import LoRALinearLayer
+from ddiffusers.models.attention import Attention
+from ddiffusers.utils import USE_PEFT_BACKEND
 from typing import Optional
 
 from einops import rearrange

genphoto/models/motion_module.py

@@ -5,9 +5,9 @@ import torch
 import torch.nn.functional as F
 from torch import nn
 
-from diffusers.utils import BaseOutput
-from diffusers.models.attention_processor import Attention
-from diffusers.models.attention import FeedForward
+from ddiffusers.utils import BaseOutput
+from ddiffusers.models.attention_processor import Attention
+from ddiffusers.models.attention import FeedForward
 
 from typing import Dict, Any
 from genphoto.models.resnet import InflatedGroupNorm

genphoto/models/resnet.py

@@ -8,9 +8,9 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 
-from diffusers.models.activations import get_activation
-from diffusers.models.normalization import AdaGroupNorm
-from diffusers.models.attention_processor import SpatialNorm
+from ddiffusers.models.activations import get_activation
+from ddiffusers.models.normalization import AdaGroupNorm
+from ddiffusers.models.attention_processor import SpatialNorm
 
 
 class InflatedConv3d(nn.Conv2d):

genphoto/models/unet_blocks.py

@@ -3,8 +3,8 @@
 import torch
 from torch import nn
 from einops import rearrange, repeat
-from diffusers.models.resnet import Downsample2D, Upsample2D, ResnetBlock2D
-from diffusers.models.transformer_2d import Transformer2DModel
+from ddiffusers.models.resnet import Downsample2D, Upsample2D, ResnetBlock2D
+from ddiffusers.models.transformer_2d import Transformer2DModel
 
 from genphoto.models.motion_module import get_motion_module