import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("robertschultz/lora-of-myself")
prompt = "TOK in a bustling cafe"
image = pipe(prompt).images[0]lora-of-myself
Model trained with AI Toolkit by Ostris
- Prompt
- TOK in a bustling cafe
- Prompt
- TOK doing a selfie
- Prompt
- TOK sitting at a computer
Trigger words
You should use TOK 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-schnell', torch_dtype=torch.bfloat16).to('cuda')
pipeline.load_lora_weights('robertschultz/lora-of-myself', weight_name='lora-of-myself')
image = pipeline('TOK in a bustling cafe').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
- 10
Model tree for robertschultz/lora-of-myself
Base model
black-forest-labs/FLUX.1-schnell