Instructions to use optima-tools-ch/image-medium with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use optima-tools-ch/image-medium with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("optima-tools-ch/image-medium", 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
- Local Apps Settings
- Draw Things
- DiffusionBee
Copie du paquet image.medium : fichiers identiques à black-forest-labs/FLUX.2-klein-4B
b9f426d verified Download transformer/config.json from optima-tools-ch/image-medium: direct link, hf CLI and curl.
- Browser
- Download file 541 Bytes
-
https://huggingface.co/optima-tools-ch/image-medium/resolve/main/transformer/config.json
- Command line
-
hf download hf://optima-tools-ch/image-medium/transformer/config.json
-
curl -L -o config.json https://huggingface.co/optima-tools-ch/image-medium/resolve/main/transformer/config.json
541 Bytes
| { | |
| "_class_name": "Flux2Transformer2DModel", | |
| "_diffusers_version": "0.37.0.dev0", | |
| "_name_or_path": "/raid/yiyi/klein-4b-distilled-diffusers/transformer", | |
| "attention_head_dim": 128, | |
| "axes_dims_rope": [ | |
| 32, | |
| 32, | |
| 32, | |
| 32 | |
| ], | |
| "eps": 1e-06, | |
| "guidance_embeds": false, | |
| "in_channels": 128, | |
| "joint_attention_dim": 7680, | |
| "mlp_ratio": 3.0, | |
| "num_attention_heads": 24, | |
| "num_layers": 5, | |
| "num_single_layers": 20, | |
| "out_channels": null, | |
| "patch_size": 1, | |
| "rope_theta": 2000, | |
| "timestep_guidance_channels": 256 | |
| } | |