BiliSakura commited on
Commit
b14b625
·
verified ·
1 Parent(s): 9cb89f8

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, Sequence, Tuple, Union
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[Sequence[Sequence[int]]] = None,
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(