imagegpt-large-bf16 / README.md
N8Programs's picture
Add ImageGPT large BF16 MLX model card
b8aa22b verified
|
Raw
History Blame Contribute Delete
1.27 kB
metadata
library_name: mlx
tags:
  - imagegpt
  - vision
  - mlx
  - safetensors
  - bfloat16
  - icl-many-replication
datasets:
  - imagenet-21k
base_model: openai/imagegpt-large

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.