import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("dong625/flux_train")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]0420
A Flux LoRA trained on a local computer with Fluxgym
Trigger words
No trigger words defined.
Download model and use it with ComfyUI, AUTOMATIC1111, SD.Next, Invoke AI, Forge, etc.
Weights for this model are available in Safetensors format.
- Downloads last month
- 10
Model tree for dong625/flux_train
Base model
black-forest-labs/FLUX.1-dev