import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("JD97/Riffusion_sentiment_LoRA", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Introduce
Riffusion with LoRA, fine-tuned with Chr0my/Epidemic_music
This model was used during Naver Connect BoostCamp AI tech 4th, NLP Track
Citation
@article{Forsgren_Martiros_2022,
author = {Forsgren, Seth* and Martiros, Hayk*},
title = {{Riffusion - Stable diffusion for real-time music generation}},
url = {https://riffusion.com/about},
year = {2022}
}
- Downloads last month
- 5