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
|