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("michaeltseng/capoo")
prompt = "BUGCAPOO on a beach"
image = pipe(prompt).images[0]capoo
Model trained with AI Toolkit by Ostris
- Prompt
- BUGCAPOO on a beach
Trigger words
You should use BUGCAPOO to trigger the image generation.
Download model and use it with ComfyUI, AUTOMATIC1111, SD.Next, Invoke AI, etc.
Weights for this model are available in Safetensors format.
Download them in the Files & versions tab.
Use it with the 🧨 diffusers library
from diffusers import AutoPipelineForText2Image
import torch
pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16).to('cuda')
pipeline.load_lora_weights('michaeltseng/capoo', weight_name='capoo.safetensors')
image = pipeline('BUGCAPOO on a beach').images[0]
image.save("my_image.png")
For more details, including weighting, merging and fusing LoRAs, check the documentation on loading LoRAs in diffusers
- Downloads last month
- 11
Model tree for michaeltseng/capoo
Base model
black-forest-labs/FLUX.1-dev