How to use AzhureRaven/rico-diffusion with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("AzhureRaven/rico-diffusion", torch_dtype=torch.bfloat16, device_map="cuda") prompt = "red Toolbar Upper Top containing Text Left Login inside and white Input Upper Top and white Input Lower Top and red Button login Upper Middle, Android UI, Medical, white background" image = pipe(prompt).images[0]
