import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("Alvinyz/lora-cube")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]LoRA text2image fine-tuning - Alvinyz/lora-cube
This model is a just small exercise during my study of lora training which is automatically saved here. It is far from perfect (may be updated or may not be updated). These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were fine-tuned on the None dataset. You can find some example images in the following.
- Downloads last month
- 3
Model tree for Alvinyz/lora-cube
Base model
runwayml/stable-diffusion-v1-5