Unverified Commit 1d9a6a81 authored by Jianqi Pan's avatar Jianqi Pan Committed by GitHub
Browse files

🔧 chore: use modeling_outputs.Transformer2DModelOutput (#8436)

* 🔧 chore: use modeling_outputs.Transformer2DModelOutput

* 🔧 chore: isort

* 🔧

 chore: isort

* style

---------
Co-authored-by: default avatarsayakpaul <spsayakpaul@gmail.com>
parent 4e0984db
...@@ -15,7 +15,8 @@ from typing import Optional ...@@ -15,7 +15,8 @@ from typing import Optional
from torch import nn from torch import nn
from .transformer_2d import Transformer2DModel, Transformer2DModelOutput from ..modeling_outputs import Transformer2DModelOutput
from .transformer_2d import Transformer2DModel
class DualTransformer2DModel(nn.Module): class DualTransformer2DModel(nn.Module):
......
...@@ -9,8 +9,8 @@ from ...models import ModelMixin ...@@ -9,8 +9,8 @@ from ...models import ModelMixin
from ...models.attention import FeedForward from ...models.attention import FeedForward
from ...models.attention_processor import Attention from ...models.attention_processor import Attention
from ...models.embeddings import TimestepEmbedding, Timesteps, get_2d_sincos_pos_embed from ...models.embeddings import TimestepEmbedding, Timesteps, get_2d_sincos_pos_embed
from ...models.modeling_outputs import Transformer2DModelOutput
from ...models.normalization import AdaLayerNorm from ...models.normalization import AdaLayerNorm
from ...models.transformers.transformer_2d import Transformer2DModelOutput
from ...utils import logging from ...utils import logging
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment