mlx-community/MI-GAN-512-places2-fp16

MI-GAN mobile inpainting, 512 Places2 (fast/on-device tier).

Converted to Apple MLX (-fp16) for Apple-Silicon inference via the mlx-lama-swift Swift package (the MLXEngine imageInpaint ModelPackage / Forge Erase capability). From-scratch MLX-Swift architecture port of Picsart-AI-Research/MI-GAN; parity-locked vs the PyTorch oracle on the CPU stream (LaMa predicted max_abs 3.2e-5 · MI-GAN 5.5e-5).

Use

// Package.swift → .package(url: "https://github.com/xocialize/mlx-lama-swift", from: "0.1.0")
import LaMa   // or MIGAN
let inpainter = try LaMaInpainter.fromPretrained(weightsPath, dtype: .float16)
let filled: CGImage = inpainter(sourceCGImage, mask: maskCGImage)  // white mask = remove

Input: image + mask (white = region to remove). Output: filled image at source resolution. Weights license: mit. Port code: MIT.

Downloads last month

-

Downloads are not tracked for this model. How to track
Safetensors
Model size
7.37M params
Tensor type
F16
·
MLX
Hardware compatibility
Log In to add your hardware

Quantized

Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Collection including mlx-community/MI-GAN-512-places2-fp16