diff --git a/llms/mlx_lm/models/qwen2.py b/llms/mlx_lm/models/qwen2.py index 41ce6d4b..42abe688 100644 --- a/llms/mlx_lm/models/qwen2.py +++ b/llms/mlx_lm/models/qwen2.py @@ -19,7 +19,6 @@ class ModelArgs(BaseModelArgs): num_key_value_heads: int = None rope_theta: float = 1000000 rope_traditional: bool = False - model_type: str = None rope_scaling: Optional[Dict[str, Union[float, str]]] = None def __post_init__(self): diff --git a/llms/setup.py b/llms/setup.py index f34fbaf7..abdf21cd 100644 --- a/llms/setup.py +++ b/llms/setup.py @@ -8,7 +8,7 @@ with open(Path(__file__).parent / "mlx_lm/requirements.txt") as fid: requirements = [str(r) for r in pkg_resources.parse_requirements(fid)] setup( name="mlx-lm", - version="0.0.10", + version="0.0.11", description="LLMs on Apple silicon with MLX and the Hugging Face Hub", long_description=open("README.md", encoding="utf-8").read(), long_description_content_type="text/markdown", diff --git a/stable_diffusion/stable_diffusion/__init__.py b/stable_diffusion/stable_diffusion/__init__.py index 9a8052c5..7f2d783c 100644 --- a/stable_diffusion/stable_diffusion/__init__.py +++ b/stable_diffusion/stable_diffusion/__init__.py @@ -130,7 +130,7 @@ class StableDiffusion: # Get the latents from the input image and add noise according to the # start time. x_0, _ = self.autoencoder.encode(image[None]) - x_0 = mx.broadcast_to(x_0, [n_images] + x_0.shape[1:]) + x_0 = mx.broadcast_to(x_0, (n_images,) + x_0.shape[1:]) x_T = self.sampler.add_noise(x_0, mx.array(start_step)) # Perform the denoising loop