How to use Miical/pi05-base with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("Miical/pi05-base", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0]
537e2e3
1
2
3
4
5
6
7
8
9
10
11
{ "_class_name": "PI0Policy", "_diffusers_version": "0.34.0", "max_action_dim": 32, "max_state_dim": 32, "n_action_steps": 50, "num_steps": 10, "pi05_enabled": true, "proj_width": 1024 }
