Update live_preview_helpers.py
Browse files- live_preview_helpers.py +2 -2
live_preview_helpers.py
CHANGED
@@ -6,7 +6,7 @@ from typing import Any, Dict, List, Optional, Union
|
|
6 |
# Helper functions
|
7 |
def calculate_shift(
|
8 |
image_seq_len,
|
9 |
-
base_seq_len: int =
|
10 |
max_seq_len: int = 4096,
|
11 |
base_shift: float = 0.5,
|
12 |
max_shift: float = 1.16,
|
@@ -58,7 +58,7 @@ def flux_pipe_call_that_returns_an_iterable_of_images(
|
|
58 |
output_type: Optional[str] = "pil",
|
59 |
return_dict: bool = True,
|
60 |
joint_attention_kwargs: Optional[Dict[str, Any]] = None,
|
61 |
-
max_sequence_length: int =
|
62 |
good_vae: Optional[Any] = None,
|
63 |
):
|
64 |
height = height or self.default_sample_size * self.vae_scale_factor
|
|
|
6 |
# Helper functions
|
7 |
def calculate_shift(
|
8 |
image_seq_len,
|
9 |
+
base_seq_len: int = 512,
|
10 |
max_seq_len: int = 4096,
|
11 |
base_shift: float = 0.5,
|
12 |
max_shift: float = 1.16,
|
|
|
58 |
output_type: Optional[str] = "pil",
|
59 |
return_dict: bool = True,
|
60 |
joint_attention_kwargs: Optional[Dict[str, Any]] = None,
|
61 |
+
max_sequence_length: int = 4096,
|
62 |
good_vae: Optional[Any] = None,
|
63 |
):
|
64 |
height = height or self.default_sample_size * self.vae_scale_factor
|