how to run on ZeroGPU?

#1
by aifeifei798 - opened

how to run on ZeroGPU?

I guess use diffusers when it is ready?

import spaces
from diffusers import DiffusionPipeline

pipe = DiffusionPipeline.from_pretrained(Efficient-Large-Model/Sana_1600M_1024px_diffusers)
pipe.to('cuda')


def generate(prompt):
    return pipe(prompt).images

gr.Interface(
    fn=generate,
    inputs=gr.Text(),
    outputs=gr.Gallery(),
).launch()

But AFAIK the diffusers pipeline stuff was still "processing"
https://github.com/huggingface/diffusers/pull/9982

hope it helps

Sign up or log in to comment