Instructions to use easylearning/RoboImagine with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use easylearning/RoboImagine with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("easylearning/RoboImagine", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Download model_validation.json from easylearning/RoboImagine: direct link, hf CLI and curl.
- Browser
- Download file 3.02 kB
-
https://huggingface.co/easylearning/RoboImagine/resolve/cb4f5e2bdfbbbff2ad6caffa058d50fa21ec3ec7/model_validation.json
- Command line
-
hf download hf://easylearning/RoboImagine@cb4f5e2bdfbbbff2ad6caffa058d50fa21ec3ec7/model_validation.json
-
curl -L -o model_validation.json https://huggingface.co/easylearning/RoboImagine/resolve/cb4f5e2bdfbbbff2ad6caffa058d50fa21ec3ec7/model_validation.json
3.02 kB
| { | |
| "state": "PASS", | |
| "torch": "2.10.0", | |
| "transformers": "4.57.1", | |
| "diffusers": "0.35.2", | |
| "cases": [ | |
| { | |
| "dataset": "rt1", | |
| "variant": "main", | |
| "example": 0, | |
| "seconds": 27.928004503250122, | |
| "shape": [ | |
| 32, | |
| 256, | |
| 320, | |
| 3 | |
| ], | |
| "finite": true, | |
| "exact_tokens": true, | |
| "max_pixel_error": 0.0 | |
| }, | |
| { | |
| "dataset": "rt1", | |
| "variant": "main", | |
| "example": 1, | |
| "seconds": 27.268405437469482, | |
| "shape": [ | |
| 32, | |
| 256, | |
| 320, | |
| 3 | |
| ], | |
| "finite": true, | |
| "exact_tokens": true, | |
| "max_pixel_error": 0.0 | |
| }, | |
| { | |
| "dataset": "rt1", | |
| "variant": "sft", | |
| "example": 0, | |
| "seconds": 27.41129207611084, | |
| "shape": [ | |
| 32, | |
| 256, | |
| 320, | |
| 3 | |
| ], | |
| "finite": true | |
| }, | |
| { | |
| "dataset": "bridge", | |
| "variant": "main", | |
| "example": 0, | |
| "seconds": 27.5578773021698, | |
| "shape": [ | |
| 32, | |
| 256, | |
| 320, | |
| 3 | |
| ], | |
| "finite": true, | |
| "exact_tokens": true, | |
| "max_pixel_error": 0.0 | |
| }, | |
| { | |
| "dataset": "bridge", | |
| "variant": "main", | |
| "example": 1, | |
| "seconds": 27.526545524597168, | |
| "shape": [ | |
| 32, | |
| 256, | |
| 320, | |
| 3 | |
| ], | |
| "finite": true, | |
| "exact_tokens": true, | |
| "max_pixel_error": 0.0 | |
| }, | |
| { | |
| "dataset": "bridge", | |
| "variant": "sft", | |
| "example": 0, | |
| "seconds": 27.738982439041138, | |
| "shape": [ | |
| 32, | |
| 256, | |
| 320, | |
| 3 | |
| ], | |
| "finite": true | |
| }, | |
| { | |
| "dataset": "robocasa", | |
| "variant": "main", | |
| "example": 0, | |
| "seconds": 29.68595314025879, | |
| "shape": [ | |
| 32, | |
| 256, | |
| 320, | |
| 3 | |
| ], | |
| "finite": true, | |
| "exact_tokens": true, | |
| "max_pixel_error": 0.0 | |
| }, | |
| { | |
| "dataset": "robocasa", | |
| "variant": "main", | |
| "example": 1, | |
| "seconds": 29.466763019561768, | |
| "shape": [ | |
| 32, | |
| 256, | |
| 320, | |
| 3 | |
| ], | |
| "finite": true, | |
| "exact_tokens": true, | |
| "max_pixel_error": 0.0 | |
| }, | |
| { | |
| "dataset": "robocasa", | |
| "variant": "sft", | |
| "example": 0, | |
| "seconds": 29.789092302322388, | |
| "shape": [ | |
| 32, | |
| 256, | |
| 320, | |
| 3 | |
| ], | |
| "finite": true | |
| } | |
| ], | |
| "notebook": { | |
| "state": "PASS", | |
| "mode": "Run All with cached SHA-256-verified bundle and local pinned T5.", | |
| "report": { | |
| "dataset": "rt1", | |
| "variant": "main", | |
| "horizon": 32, | |
| "window": 6, | |
| "retain_anchor": true, | |
| "beam_width": 4, | |
| "seconds_single_unwarmed_run": 27.482906818389893, | |
| "precision": "float32", | |
| "seed": 20261011, | |
| "torch": "2.10.0", | |
| "psnr_db": 23.423784299674352, | |
| "mae": 0.037065139831221165, | |
| "reference_check": { | |
| "exact_tokens": true, | |
| "maximum_pixel_error": 0.0 | |
| } | |
| }, | |
| "mp4_bytes": 86913 | |
| } | |
| } | |