no relative

models/champ_flame_model.py → champ_flame_model.py

@@ -6,7 +6,7 @@
 
 import torch
 import torch.nn as nn
-from models.unet_2d_condition import UNet2DConditionModel
+from unet_2d_condition import UNet2DConditionModel
 import os
 import json
 import shutil

models/guidance_encoder.py → guidance_encoder.py

@@ -8,10 +8,10 @@ from diffusers.models.modeling_utils import ModelMixin
 from diffusers.utils import BaseOutput
 from dataclasses import dataclass
 
-from models.motion_module import zero_module
-from models.resnet import InflatedConv3d, InflatedGroupNorm
-from models.attention import TemporalBasicTransformerBlock
-from models.transformer_3d import Transformer3DModel
+from motion_module import zero_module
+from resnet import InflatedConv3d, InflatedGroupNorm
+from attention import TemporalBasicTransformerBlock
+from transformer_3d import Transformer3DModel
 
 
 class GuidanceEncoder(ModelMixin):

models/mutual_self_attention.py → mutual_self_attention.py

@@ -4,9 +4,9 @@ from typing import Any, Dict, Optional
 import torch
 from einops import rearrange
 
-from models.attention import TemporalBasicTransformerBlock
+from attention import TemporalBasicTransformerBlock
 
-from .attention import BasicTransformerBlock
+from attention import BasicTransformerBlock
 
 
 def torch_dfs(model: torch.nn.Module):

pipeline/pipeline_stable_diffusion.py → pipeline_stable_diffusion.py

@@ -40,7 +40,7 @@ from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionS
 ###########################
 #
 # VF
-from models.guidance_encoder import GuidanceEncoder
+from guidance_encoder import GuidanceEncoder
 import numpy as np
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name

models/transformer_3d.py → transformer_3d.py

@@ -9,7 +9,7 @@ from diffusers.utils.import_utils import is_xformers_available
 from einops import rearrange, repeat
 from torch import nn
 
-from .attention import TemporalBasicTransformerBlock
+from attention import TemporalBasicTransformerBlock
 
 
 @dataclass