Instructions to use chen-hao-chao/mdm-prime with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use chen-hao-chao/mdm-prime with Transformers:

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="chen-hao-chao/mdm-prime")

# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("chen-hao-chao/mdm-prime", device_map="auto")

Notebooks
Google Colab
Kaggle
Local Apps Settings

vLLM

How to use chen-hao-chao/mdm-prime with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "chen-hao-chao/mdm-prime"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "chen-hao-chao/mdm-prime",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker

docker model run hf.co/chen-hao-chao/mdm-prime

SGLang

How to use chen-hao-chao/mdm-prime with SGLang:

Install from pip and serve model

# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
    --model-path "chen-hao-chao/mdm-prime" \
    --host 0.0.0.0 \
    --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "chen-hao-chao/mdm-prime",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker images

docker run --gpus all \
    --shm-size 32g \
    -p 30000:30000 \
    -v ~/.cache/huggingface:/root/.cache/huggingface \
    --env "HF_TOKEN=<secret>" \
    --ipc=host \
    lmsysorg/sglang:latest \
    python3 -m sglang.launch_server \
        --model-path "chen-hao-chao/mdm-prime" \
        --host 0.0.0.0 \
        --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "chen-hao-chao/mdm-prime",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Docker Model Runner
How to use chen-hao-chao/mdm-prime with Docker Model Runner:
```
docker model run hf.co/chen-hao-chao/mdm-prime
```

chen-hao-chao commited on Jun 6, 2025

Commit

103d177

verified ·

1 Parent(s): 8be1367

Upload folder using huggingface_hub

Browse files

Files changed (14) hide show

image/cifar10/results_mdm_cifar10/checkpoint-2499.pth +3 -0
image/cifar10/results_prime_l2_cifar10/checkpoint-4249.pth +3 -0
image/cifar10/results_prime_l3_cifar10/checkpoint-3499.pth +3 -0
image/cifar10/results_prime_l4_cifar10/checkpoint-2499.pth +3 -0
image/cifar10/results_prime_l6_cifar10/checkpoint-1499.pth +3 -0
image/cifar10/results_prime_l8_cifar10/checkpoint-1499.pth +3 -0
image/cifar10/results_prime_supertoken_cifar10/checkpoint-2249.pth +3 -0
image/imagenet32/results_mdm_imagenet32/checkpoint-674.pth +3 -0
image/imagenet32/results_prime_l2_imagenet32/checkpoint-974.pth +3 -0
image/imagenet32/results_prime_l3_imagenet32/checkpoint-1049.pth +3 -0
image/imagenet32/results_prime_l4_imagenet32/checkpoint-899.pth +3 -0
image/imagenet32/results_prime_l6_imagenet32/checkpoint-674.pth +3 -0
image/imagenet32/results_prime_l8_imagenet32/checkpoint-649.pth +3 -0
image/imagenet32/results_prime_supertoken_imagenet32/checkpoint-599.pth +3 -0

image/cifar10/results_mdm_cifar10/checkpoint-2499.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:669340ebd097321899c74d48d9fa7327bfdfd6fb8d98e37cda424479885d5cde
+size 1824659259

image/cifar10/results_prime_l2_cifar10/checkpoint-4249.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:37ca880aab0302dbffbbbeb7818d77bd619bd528284c298a8f7e2553f9fd61bb
+size 1824153595

image/cifar10/results_prime_l3_cifar10/checkpoint-3499.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:9ae64f9fd7224c5fac34182608e46a676128311c7d1af4ac367004ad1f4ce6bd
+size 1824144507

image/cifar10/results_prime_l4_cifar10/checkpoint-2499.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:ae5bd9f358f855b01ec75d2caad8ea529523bdf6af572aa926ae9fe749d5c320
+size 1824142203

image/cifar10/results_prime_l6_cifar10/checkpoint-1499.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:940279f3c5e181415477e254297044bd8ab572d6beee7d680c7fcf9bc0da24ba
+size 1824141435

image/cifar10/results_prime_l8_cifar10/checkpoint-1499.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:92468e6cb94ab4d1d0091cd73f06fe0387e32e41dfdc6748724526889d48d844
+size 1824140923

image/cifar10/results_prime_supertoken_cifar10/checkpoint-2249.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:6b63a9a12f5ea3480dd33769ce576d75f1b40bd197d55db03e48910cea65f439
+size 2141670331

image/imagenet32/results_mdm_imagenet32/checkpoint-674.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:78d6c5c87428b0d2da9cb1bc788a11beb6ebdc7614c4d3bb9b1c576704765dfc
+size 1824658064

image/imagenet32/results_prime_l2_imagenet32/checkpoint-974.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:278ad0dd7627329bd133f47a8f3288ab321b3e9bcae406843e742730c4d746bd
+size 1824151952

image/imagenet32/results_prime_l3_imagenet32/checkpoint-1049.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:a98baf261e584d78e02c21a880d98d89a6f78fe938a69f45792b78473a6f7f42
+size 1824145147

image/imagenet32/results_prime_l4_imagenet32/checkpoint-899.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:e0b7231c7c523523cd2434b99fa42d6ef5154dde0d51519c2461f0b38dc8c6b6
+size 1824140688

image/imagenet32/results_prime_l6_imagenet32/checkpoint-674.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:d4d75a066f050964ee8db857abb399123b6445cc0820e9fb332963e0a055755e
+size 1824139920

image/imagenet32/results_prime_l8_imagenet32/checkpoint-649.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:05853b5cdcb499b97e6460224c55281d56cee0bf1a037a2257e0dfb060dbcc74
+size 1824139280

image/imagenet32/results_prime_supertoken_imagenet32/checkpoint-599.pth ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:190e281d644510c06d7d0381fbc703c98f02c1b9eb3113e79c13e059986e6f15
+size 2141668240