File size: 538 Bytes
5c7905f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
import torch
def dummy(images, **kwargs):
return images, False
async def Gimager(prompt):
model_id = "SG161222/Realistic_Vision_V1.4"
scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
pipe = await StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
pipe = pipe.to("cuda")
pipe.safety_checker = dummy
image = await pipe(prompt).images[0]
return image
|