Add files

app.py

@@ -222,6 +222,7 @@ def remove_decoupled_file(path, exta='.mp4', extb='.json'):
     [os.remove(osp.join(path, tagi+exta)) for tagi in tag_a_extra]
     [os.remove(osp.join(path, tagi+extb)) for tagi in tag_b_extra]
 
+@spaces.GPU()
 @torch.no_grad()
 def t2i_core(model, xt, emb, nemb, step=30, cfg_scale=7.5, return_list=False):
     from nulltxtinv_wrapper import diffusion_step, latent2image
@@ -319,6 +320,7 @@ class wrapper(object):
     # main #
     ########
 
+    @spaces.GPU()
     def ddiminv(self, img, cfgdict):
         txt, step, cfg_scale = cfgdict['txt'], cfgdict['step'], cfgdict['cfg_scale']
         from nulltxtinv_wrapper import NullInversion
@@ -333,6 +335,7 @@ class wrapper(object):
             'xt': xt, 'emb': emb, 'nemb': nemb,}
         return data
 
+    @spaces.GPU()
     def nullinv_or_loadcache(self, img, cfgdict, force_reinvert=False):
         hash = hash_pilim(img) + "--" + hash_cfgdict(cfgdict)
         cdir = self.cache_inverse_folder
@@ -373,6 +376,7 @@ class wrapper(object):
                 'xt' : xt, 'emb' : emb, 'nemb' : nemb,}
             return data
 
+    @spaces.GPU()
     def nullinvdual_or_loadcachedual(self, img0, img1, cfgdict, force_reinvert=False):
         hash = hash_pilim(img0) + "--" + hash_pilim(img1) + "--" + hash_cfgdict(cfgdict)
         cdir = self.cache_inverse_folder
@@ -437,6 +441,7 @@ class wrapper(object):
                 'xt' : xt1, 'emb' : emb1, 'nemb' : nemb,}
             return data0, data1
 
+    @spaces.GPU()
     def image_inversion(
             self, img, txt, 
             cfg_scale, step, 
@@ -463,6 +468,7 @@ class wrapper(object):
             
         return img_inv
 
+    @spaces.GPU()
     def image_editing(
         self, img, txt_0, txt_1,
         cfg_scale, step, thresh, 
@@ -486,6 +492,7 @@ class wrapper(object):
         
         return img_edited
 
+    @spaces.GPU()
     def general_interpolation(
             self, xset0, xset1,
             cfg_scale, step, tlist,):