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("Dharneesh/test1")
prompt = "sunglasses at marina beach"
image = pipe(prompt).images[0]test1
Model trained with AI Toolkit by Ostris
- Prompt
- sunglasses at marina beach
Trigger words
No trigger words defined.
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('Dharneesh/test1', weight_name='test1.safetensors')
image = pipeline('sunglasses at marina 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
- 22
Model tree for Dharneesh/test1
Base model
black-forest-labs/FLUX.1-schnell