BAAI/Infinity-Instruct
Viewer • Updated • 21.9M • 2.92k • 727
How to use RustyDanger/Weapon_Intel with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("RustyDanger/Weapon_Intel", dtype=torch.bfloat16, device_map="cuda")
prompt = "Where do I live?"
image = pipe(prompt).images[0]