Instructions to use Balta10/fire with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use Balta10/fire with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline from diffusers.utils import load_image # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("Balta10/fire", dtype=torch.bfloat16, device_map="cuda") prompt = "Turn this cat into a dog" input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png") image = pipe(image=input_image, prompt=prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Download transformer/config.json from Balta10/fire: direct link, hf CLI and curl.
- Browser
- Download file 1.07 kB
-
https://huggingface.co/Balta10/fire/resolve/main/transformer/config.json
- Command line
-
hf download hf://Balta10/fire/transformer/config.json
-
curl -L -o config.json https://huggingface.co/Balta10/fire/resolve/main/transformer/config.json
1.07 kB
| { | |
| "_class_name": "QwenImageTransformer2DModel", | |
| "_diffusers_version": "0.37.0.dev0", | |
| "_name_or_path": "/home/user/.cache/huggingface/hub/models--FireRedTeam--FireRed-Image-Edit-1.0/snapshots/0aea9d520c801c9f4b691cc92f736a7a8628e6a2/transformer", | |
| "attention_head_dim": 128, | |
| "axes_dims_rope": [ | |
| 16, | |
| 56, | |
| 56 | |
| ], | |
| "guidance_embeds": false, | |
| "in_channels": 64, | |
| "joint_attention_dim": 3584, | |
| "num_attention_heads": 24, | |
| "num_layers": 60, | |
| "out_channels": 16, | |
| "patch_size": 2, | |
| "quantization_config": { | |
| "_load_in_4bit": false, | |
| "_load_in_8bit": true, | |
| "bnb_4bit_compute_dtype": "float32", | |
| "bnb_4bit_quant_storage": "uint8", | |
| "bnb_4bit_quant_type": "fp4", | |
| "bnb_4bit_use_double_quant": false, | |
| "llm_int8_enable_fp32_cpu_offload": false, | |
| "llm_int8_has_fp16_weight": false, | |
| "llm_int8_skip_modules": null, | |
| "llm_int8_threshold": 6.0, | |
| "load_in_4bit": false, | |
| "load_in_8bit": true, | |
| "quant_method": "bitsandbytes" | |
| }, | |
| "use_additional_t_cond": false, | |
| "use_layer3d_rope": false, | |
| "zero_cond_t": false | |
| } | |