starhopp3r/TinyChat
Viewer • Updated • 1M • 226 • 19
How to use Roman190928/12MLangModel15000 with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Roman190928/12MLangModel15000", dtype=torch.bfloat16, device_map="cuda")
prompt = "My name is Julien and I like to"
image = pipe(prompt).images[0]Base model
Roman190928/12MTRANSFORMER