Skip to content

Commit da62500

Browse files
authored
fix: mlx: Wrong type annotation for latent_size
Pyright: Argument of type "tuple[Literal[64], Literal[64]]" cannot be assigned to parameter "latent_size" of type "Tuple[int]" in function "generate_image"   "tuple[Literal[64], Literal[64]]" is not assignable to "Tuple[int]"     Tuple size mismatch; expected 1 but received 2
1 parent 2fc6f7c commit da62500

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/src/diffusionkit/mlx/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -256,7 +256,7 @@ def denoise_latents(
256256
pooled_conditioning,
257257
num_steps: int = 2,
258258
cfg_weight: float = 0.0,
259-
latent_size: Tuple[int] = (64, 64),
259+
latent_size: Tuple[int, int] = (64, 64),
260260
seed=None,
261261
image_path: Optional[str] = None,
262262
denoise: float = 1.0,
@@ -297,7 +297,7 @@ def generate_image(
297297
num_steps: int = 2,
298298
cfg_weight: float = 0.0,
299299
negative_text: str = "",
300-
latent_size: Tuple[int] = (64, 64),
300+
latent_size: Tuple[int, int] = (64, 64),
301301
seed=None,
302302
verbose: bool = True,
303303
image_path: Optional[str] = None,

0 commit comments

Comments
 (0)