Alexandros Graikos
commited on
Commit
·
c4ff05a
1
Parent(s):
9e039a3
emtpy
Browse files- pipeline.py +1 -1
- pixcell_transformer_2d.py +1 -1
pipeline.py
CHANGED
|
@@ -32,7 +32,7 @@ from diffusers.utils import (
|
|
| 32 |
from diffusers.utils.torch_utils import randn_tensor
|
| 33 |
from diffusers.pipelines.pipeline_utils import DiffusionPipeline, ImagePipelineOutput
|
| 34 |
|
| 35 |
-
from pixcell_transformer_2d import PixCellTransformer2DModel
|
| 36 |
|
| 37 |
|
| 38 |
# TODO:
|
|
|
|
| 32 |
from diffusers.utils.torch_utils import randn_tensor
|
| 33 |
from diffusers.pipelines.pipeline_utils import DiffusionPipeline, ImagePipelineOutput
|
| 34 |
|
| 35 |
+
from .pixcell_transformer_2d import PixCellTransformer2DModel
|
| 36 |
|
| 37 |
|
| 38 |
# TODO:
|
pixcell_transformer_2d.py
CHANGED
|
@@ -25,7 +25,7 @@ from diffusers.models.modeling_outputs import Transformer2DModelOutput
|
|
| 25 |
from diffusers.models.modeling_utils import ModelMixin
|
| 26 |
from diffusers.models.normalization import AdaLayerNormSingle
|
| 27 |
|
| 28 |
-
from embeddings_pixcell import PixcellUNIProjection, UNIPosEmbed
|
| 29 |
|
| 30 |
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
| 31 |
|
|
|
|
| 25 |
from diffusers.models.modeling_utils import ModelMixin
|
| 26 |
from diffusers.models.normalization import AdaLayerNormSingle
|
| 27 |
|
| 28 |
+
from .embeddings_pixcell import PixcellUNIProjection, UNIPosEmbed
|
| 29 |
|
| 30 |
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
| 31 |
|