Spaces:

akameswa
/

DiffusionDemo

Running on Zero

akameswa commited on Apr 10

Commit

29a0427

•

1 Parent(s): 7fd88e0

Update src/pipelines/perturbations.py

Files changed (1) hide show

src/pipelines/perturbations.py CHANGED Viewed

@@ -1,10 +1,11 @@
 import torch
 import numpy as np
 import gradio as gr
 from src.util.base import *
 from src.util.params import *
 def display_perturb_images(
     prompt,
     seed,

 import torch
+import spaces
 import numpy as np
 import gradio as gr
 from src.util.base import *
 from src.util.params import *
+@spaces.GPU(enable_queue=True)
 def display_perturb_images(
     prompt,
     seed,