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

MI-GAN mobile inpainting, 256 Places2.

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
6.29M 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-256-places2-fp16