Spaces:

softwareweaver
/

ImageGenPlus

Running on Zero

softwareweaver commited on Nov 7

Commit

2b169b3

1 Parent(s): 1cc732c

zero gpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 import numpy as np
 import random
-# import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
 import torch
@@ -29,7 +29,7 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
-# @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
     prompt,
     negative_prompt,

 import numpy as np
 import random
+from spaces import ZeroGPU
 from diffusers import DiffusionPipeline
 import torch
 MAX_IMAGE_SIZE = 1024
+@ZeroGPU
 def infer(
     prompt,
     negative_prompt,