Spaces:

linoyts
/

edit_friendly_ddpm_inversion

Running on A10G

linoyts HF staff commited on 29 days ago

Commit

2353090

•

1 Parent(s): d2a6d44

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,9 @@ from utils import *
 from inversion_utils import *
 from torch import autocast, inference_mode
 import re
 def randomize_seed_fn(seed, randomize_seed):
     if randomize_seed:
         seed = random.randint(0, np.iinfo(np.int32).max)

 from inversion_utils import *
 from torch import autocast, inference_mode
 import re
+import spaces
+@spaces.GPU
 def randomize_seed_fn(seed, randomize_seed):
     if randomize_seed:
         seed = random.randint(0, np.iinfo(np.int32).max)