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("Rakoo04/newton-dog")
prompt = "A person in a bustling cafe Nwtn"
image = pipe(prompt).images[0]newton-dog
Model trained with AI Toolkit by Ostris
- Prompt
- A person in a bustling cafe Nwtn
Trigger words
You should use Nwtn 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('Rakoo04/newton-dog', weight_name='newton-dog')
image = pipeline('A person in a bustling cafe Nwtn').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
- 12
Model tree for Rakoo04/newton-dog
Base model
black-forest-labs/FLUX.1-schnell