AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

🔧 chore: use modeling_outputs.Transformer2DModelOutput (#8436)

Browse Source 
* 🔧 chore: use modeling_outputs.Transformer2DModelOutput

* 🔧 chore: isort

* 🔧 chore: isort

* style

---------

Co-authored-by: sayakpaul <spsayakpaul@gmail.com>
This commit is contained in:
Jianqi Pan
2024-06-10 20:11:41 +09:00
committed by GitHub
parent 4e0984db6c
commit 1d9a6a81b9
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/diffusers/models/transformers/dual_transformer_2d.py
View File

@@ -15,7 +15,8 @@ from typing import Optional
from torch import nn
from .transformer_2d import Transformer2DModel, Transformer2DModelOutput
from ..modeling_outputs import Transformer2DModelOutput
from .transformer_2d import Transformer2DModel
class DualTransformer2DModel(nn.Module):

2
src/diffusers/pipelines/unidiffuser/modeling_uvit.py
View File

@@ -9,8 +9,8 @@ from ...models import ModelMixin
from ...models.attention import FeedForward
from ...models.attention_processor import Attention
from ...models.embeddings import TimestepEmbedding, Timesteps, get_2d_sincos_pos_embed
from ...models.modeling_outputs import Transformer2DModelOutput
from ...models.normalization import AdaLayerNorm
from ...models.transformers.transformer_2d import Transformer2DModelOutput
from ...utils import logging