Update app.py

app.py

@@ -32,12 +32,11 @@ from diffusers import DiffusionPipeline, AnimateDiffPipeline, MotionAdapter, Eul
 #import jax
 #import jax.numpy as jnp
 from numba import njit as cpu1, jit as cpu2
-from numba.cuda import njit as cuda1, jit as cuda2
+from numba.cuda import jit as cuda
 
 # optimization:
 
-# @cuda1(cache=True)
-# @cuda2(cache=True)
+# @cuda(cache=True)
 # @cpu1(cache=True,nopython=True,parallel=True)
 # @cpu2(cache=True,nopython=True,parallel=True)
 # @cpu1(cache=True)
@@ -127,8 +126,7 @@ function custom(){
 
 # functionality
 
-@cuda1(cache=True)
-# @cuda2(cache=True)
+@cuda(cache=True)
 # @cpu1(cache=True,nopython=True,parallel=True)
 # @cpu2(cache=True,nopython=True,parallel=True)
 # @cpu1(cache=True)
@@ -147,8 +145,7 @@ def run(cmd, assert_success=False, capture_output=False, env=None, dry_run=False
 
     return result
     
-@cuda1(cache=True)
-# @cuda2(cache=True)
+@cuda(cache=True)
 # @cpu1(cache=True,nopython=True,parallel=True)
 # @cpu2(cache=True,nopython=True,parallel=True)
 # @cpu1(cache=True)
@@ -194,8 +191,7 @@ def translate(text,lang):
     print(ret)
     return ret
     
-@cuda1(cache=True)
-# @cuda2(cache=True)
+@cuda(cache=True)
 # @cpu1(cache=True,nopython=True,parallel=True)
 # @cpu2(cache=True,nopython=True,parallel=True)
 # @cpu1(cache=True)
@@ -204,8 +200,7 @@ def generate_random_string(length):
     characters = string.ascii_letters + string.digits
     return ''.join(random.choice(characters) for _ in range(length))
     
-@cuda1(cache=True)
-# @cuda2(cache=True)
+@cuda(cache=True)
 # @cpu1(cache=True,nopython=True,parallel=True)
 # @cpu2(cache=True,nopython=True,parallel=True)
 # @cpu1(cache=True)
@@ -247,8 +242,7 @@ def Piper(image,positive,negative,motion):
         num_frames=(fps*time)
     )
 
-@cuda1(cache=True)
-# @cuda2(cache=True)
+@cuda(cache=True)
 # @cpu1(cache=True,nopython=True,parallel=True)
 # @cpu2(cache=True,nopython=True,parallel=True)
 # @cpu1(cache=True)
@@ -274,8 +268,7 @@ def infer(pm):
         export_to_gif(out.frames[0],name,fps=fps)
         return name
 
-@cuda1(cache=True)
-# @cuda2(cache=True)
+@cuda(cache=True)
 # @cpu1(cache=True,nopython=True,parallel=True)
 # @cpu2(cache=True,nopython=True,parallel=True)
 # @cpu1(cache=True)
@@ -294,8 +287,7 @@ def handle(i,m,p1,p2,*result):
         ret.append(infer,pm)
     return ret
 
-@cuda1(cache=True)
-# @cuda2(cache=True)
+@cuda(cache=True)
 # @cpu1(cache=True,nopython=True,parallel=True)
 # @cpu2(cache=True,nopython=True,parallel=True)
 # @cpu1(cache=True)
@@ -353,8 +345,7 @@ def ui():
         )
         demo.queue().launch()
 
-@cuda1(cache=True)
-# @cuda2(cache=True)
+@cuda(cache=True)
 # @cpu1(cache=True,nopython=True,parallel=True)
 # @cpu2(cache=True,nopython=True,parallel=True)
 # @cpu1(cache=True)