ImageGPT Large BF16 MLX

This repository contains the local bfloat16/MLX conversion of openai/imagegpt-large used by the ImageGPT runs in the ICLManyReplication artifact for "Many Next-Token Predictors are In-Context Learners".

The upload includes:

model.safetensors: converted bfloat16 model weights
config.json: ImageGPT model configuration with the local MLX model file hook
imagegpt_mlx_lm.py: local MLX model implementation used by the replication harness
preprocessor_config.json: original ImageGPT preprocessing configuration
conversion_metadata.json: source path, dtype, tensor count, and conversion notes

The conversion metadata records openai/imagegpt-large weights converted from the original PyTorch checkpoint into a bfloat16 safetensors artifact.

Intended Use

This checkpoint is intended as a drop-in artifact for reproducing the ImageGPT principal-result row in ICLManyReplication. For general ImageGPT usage, see the upstream openai/imagegpt-large model card.

Downloads last month: -

Safetensors

Model size

1B params

Tensor type

BF16

MLX

Hardware compatibility

Quantized

Inference Providers NEW

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

Model tree for N8Programs/imagegpt-large-bf16

Base model

openai/imagegpt-large

Finetuned

(1)

this model