Skip to content

Commit cf61b0a

Browse files
committed
make style, make quality and make fix-copies
1 parent 87b5383 commit cf61b0a

5 files changed

Lines changed: 36 additions & 28 deletions

File tree

src/diffusers/models/transformers/transformer_joyimage.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -330,8 +330,6 @@ def forward(
330330
return hidden_states, encoder_hidden_states
331331

332332

333-
# ---------------------------------------------------------------------------
334-
# Copied from diffusers.models.transformers.transformer_wan.WanTimeTextImageEmbedding
335333
class JoyImageTimeTextImageEmbedding(nn.Module):
336334
def __init__(
337335
self,
@@ -566,9 +564,7 @@ def forward(
566564
# main loop
567565
for block in self.double_blocks:
568566
if torch.is_grad_enabled() and self.gradient_checkpointing:
569-
img, txt = self._gradient_checkpointing_func(
570-
block, img, txt, vec, (vis_freqs, txt_freqs)
571-
)
567+
img, txt = self._gradient_checkpointing_func(block, img, txt, vec, (vis_freqs, txt_freqs))
572568
else:
573569
img, txt = block(
574570
hidden_states=img,
@@ -592,4 +588,3 @@ def forward(
592588
if not return_dict:
593589
return (img,)
594590
return Transformer2DModelOutput(sample=img)
595-

src/diffusers/pipelines/joyimage/image_processor.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515
import math
1616
from typing import Tuple
1717

18-
import torch
1918
from PIL import Image
2019

2120
from ...configuration_utils import register_to_config

src/diffusers/pipelines/joyimage/pipeline_joyimage_edit.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
import torch
66
from PIL import Image
77
from transformers import (
8-
AutoProcessor,
98
Qwen2Tokenizer,
109
Qwen3VLForConditionalGeneration,
1110
Qwen3VLProcessor,

src/diffusers/utils/dummy_pt_objects.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1290,7 +1290,7 @@ def from_pretrained(cls, *args, **kwargs):
12901290
requires_backends(cls, ["torch"])
12911291

12921292

1293-
class JoyImageEditTransformer3DModel(metaclass=DummyObject):
1293+
class HunyuanVideo15Transformer3DModel(metaclass=DummyObject):
12941294
_backends = ["torch"]
12951295

12961296
def __init__(self, *args, **kwargs):
@@ -1305,7 +1305,7 @@ def from_pretrained(cls, *args, **kwargs):
13051305
requires_backends(cls, ["torch"])
13061306

13071307

1308-
class HunyuanVideo15Transformer3DModel(metaclass=DummyObject):
1308+
class HunyuanVideoFramepackTransformer3DModel(metaclass=DummyObject):
13091309
_backends = ["torch"]
13101310

13111311
def __init__(self, *args, **kwargs):
@@ -1320,7 +1320,7 @@ def from_pretrained(cls, *args, **kwargs):
13201320
requires_backends(cls, ["torch"])
13211321

13221322

1323-
class HunyuanVideoFramepackTransformer3DModel(metaclass=DummyObject):
1323+
class HunyuanVideoTransformer3DModel(metaclass=DummyObject):
13241324
_backends = ["torch"]
13251325

13261326
def __init__(self, *args, **kwargs):
@@ -1335,7 +1335,7 @@ def from_pretrained(cls, *args, **kwargs):
13351335
requires_backends(cls, ["torch"])
13361336

13371337

1338-
class HunyuanVideoTransformer3DModel(metaclass=DummyObject):
1338+
class I2VGenXLUNet(metaclass=DummyObject):
13391339
_backends = ["torch"]
13401340

13411341
def __init__(self, *args, **kwargs):
@@ -1350,7 +1350,7 @@ def from_pretrained(cls, *args, **kwargs):
13501350
requires_backends(cls, ["torch"])
13511351

13521352

1353-
class I2VGenXLUNet(metaclass=DummyObject):
1353+
class JoyImageEditTransformer3DModel(metaclass=DummyObject):
13541354
_backends = ["torch"]
13551355

13561356
def __init__(self, *args, **kwargs):

src/diffusers/utils/dummy_torch_and_transformers_objects.py

Lines changed: 30 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1907,6 +1907,36 @@ def from_pretrained(cls, *args, **kwargs):
19071907
requires_backends(cls, ["torch", "transformers"])
19081908

19091909

1910+
class JoyImageEditPipeline(metaclass=DummyObject):
1911+
_backends = ["torch", "transformers"]
1912+
1913+
def __init__(self, *args, **kwargs):
1914+
requires_backends(self, ["torch", "transformers"])
1915+
1916+
@classmethod
1917+
def from_config(cls, *args, **kwargs):
1918+
requires_backends(cls, ["torch", "transformers"])
1919+
1920+
@classmethod
1921+
def from_pretrained(cls, *args, **kwargs):
1922+
requires_backends(cls, ["torch", "transformers"])
1923+
1924+
1925+
class JoyImageEditPipelineOutput(metaclass=DummyObject):
1926+
_backends = ["torch", "transformers"]
1927+
1928+
def __init__(self, *args, **kwargs):
1929+
requires_backends(self, ["torch", "transformers"])
1930+
1931+
@classmethod
1932+
def from_config(cls, *args, **kwargs):
1933+
requires_backends(cls, ["torch", "transformers"])
1934+
1935+
@classmethod
1936+
def from_pretrained(cls, *args, **kwargs):
1937+
requires_backends(cls, ["torch", "transformers"])
1938+
1939+
19101940
class Kandinsky3Img2ImgPipeline(metaclass=DummyObject):
19111941
_backends = ["torch", "transformers"]
19121942

@@ -2567,21 +2597,6 @@ def from_pretrained(cls, *args, **kwargs):
25672597
requires_backends(cls, ["torch", "transformers"])
25682598

25692599

2570-
class JoyImageEditPipeline(metaclass=DummyObject):
2571-
_backends = ["torch", "transformers"]
2572-
2573-
def __init__(self, *args, **kwargs):
2574-
requires_backends(self, ["torch", "transformers"])
2575-
2576-
@classmethod
2577-
def from_config(cls, *args, **kwargs):
2578-
requires_backends(cls, ["torch", "transformers"])
2579-
2580-
@classmethod
2581-
def from_pretrained(cls, *args, **kwargs):
2582-
requires_backends(cls, ["torch", "transformers"])
2583-
2584-
25852600
class Lumina2Pipeline(metaclass=DummyObject):
25862601
_backends = ["torch", "transformers"]
25872602

0 commit comments

Comments
 (0)