How to use tiny-random/moss-image with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("tiny-random/moss-image", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0]
7ff060e f84967c 7ff060e
1
2
3
4
5
6
7
8
9
10
11
12
13
{ "_class_name": "MOSSImageTransformer", "_diffusers_version": "0.40.0.dev0", "dim": 64, "ffn_dim": 128, "freq_dim": 32, "latent_channels": 128, "num_heads": 2, "num_layers": 4, "rope_base": 10000.0, "text_dim": 32 }
