Skip to content

Commit ad40575

Browse files
Update pipeline_cogview4.py
1 parent 030a467 commit ad40575

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/diffusers/pipelines/cogview4/pipeline_cogview4.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,12 +55,11 @@
5555
"""
5656

5757

58-
def time_shift(self, mu: float, shift_sigma: float, sigmas: torch.Tensor):
58+
def time_shift(mu: float, shift_sigma: float, sigmas: torch.Tensor):
5959
return mu / (mu + (1 / sigmas - 1) ** shift_sigma)
6060

6161

6262
def calculate_shift(
63-
self,
6463
image_seq_len,
6564
base_seq_len: int = 256,
6665
):

0 commit comments

Comments
 (0)