Update all files for BitDance-14B-16x-diffusers
Browse files
bitdance_diffusers/pipeline_bitdance.py
CHANGED
|
@@ -1,7 +1,5 @@
|
|
| 1 |
-
from __future__ import annotations
|
| 2 |
-
|
| 3 |
from contextlib import nullcontext
|
| 4 |
-
from typing import List, Optional,
|
| 5 |
|
| 6 |
import torch
|
| 7 |
from einops import rearrange
|
|
@@ -34,7 +32,7 @@ class BitDanceDiffusionPipeline(DiffusionPipeline):
|
|
| 34 |
autoencoder,
|
| 35 |
diffusion_head,
|
| 36 |
projector,
|
| 37 |
-
supported_image_sizes: Optional[
|
| 38 |
) -> None:
|
| 39 |
super().__init__()
|
| 40 |
self.register_modules(
|
|
|
|
|
|
|
|
|
|
| 1 |
from contextlib import nullcontext
|
| 2 |
+
from typing import List, Optional, Tuple, Union
|
| 3 |
|
| 4 |
import torch
|
| 5 |
from einops import rearrange
|
|
|
|
| 32 |
autoencoder,
|
| 33 |
diffusion_head,
|
| 34 |
projector,
|
| 35 |
+
supported_image_sizes: Optional[List[List[int]]] = None,
|
| 36 |
) -> None:
|
| 37 |
super().__init__()
|
| 38 |
self.register_modules(
|