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("Makooooooooo/JUDGE_3000STEP")
prompt = "TCJUDGE02, TCJUDGE03, TCJUDGE04, TCJUDGE05, TCJUDGE06"
image = pipe(prompt).images[0]JUDGE_3000STEP
Model description
Trigger words
You should use TCJUDGE02, TCJUDGE03, TCJUDGE04, TCJUDGE05, TCJUDGE06 to trigger the image generation.
Download model
Weights for this model are available in Safetensors format.
Download them in the Files & versions tab.
Training at fal.ai
Training was done using fal.ai/models/fal-ai/flux-lora-fast-training.
- Downloads last month
- 1
Model tree for Makooooooooo/JUDGE_3000STEP
Base model
black-forest-labs/FLUX.1-dev