Spaces:

Fucius
/

OMG

Runtime error

Fucius commited on Mar 19

Commit

7b43b38

•

1 Parent(s): 44a822d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,6 +6,7 @@ import torch
 pipe = DiffusionPipeline.from_pretrained("Fucius/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16")
 pipe.to('cuda')
 @spaces.GPU
 def generate(prompt):
     return pipe(prompt).images

 pipe = DiffusionPipeline.from_pretrained("Fucius/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16")
 pipe.to('cuda')
 @spaces.GPU
 def generate(prompt):
     return pipe(prompt).images