Yaron Koresh commited on
Commit
e13a4c7
·
verified ·
1 Parent(s): 6f6a061

Update code.py

Browse files
Files changed (1) hide show
  1. code.py +9 -4
code.py CHANGED
@@ -405,14 +405,19 @@ def pre():
405
  pipe.enable_vae_slicing()
406
  pipe.enable_free_init(method="butterworth", use_fast_sampling=fast)
407
 
408
- # entry
409
-
 
 
 
410
  def entry():
411
-
412
  os.chdir(os.path.abspath(os.path.dirname(__file__)))
413
  mp.set_start_method("spawn", force=True)
414
-
415
  pre[32,32]()
416
  ui[32,32]()
417
 
 
 
 
 
418
  # end
 
405
  pipe.enable_vae_slicing()
406
  pipe.enable_free_init(method="butterworth", use_fast_sampling=fast)
407
 
408
+ @gpu(cache=True)
409
+ # @cpu1(cache=True,nopython=True,parallel=True)
410
+ # @cpu2(cache=True,nopython=True,parallel=True)
411
+ # @cpu1(cache=True)
412
+ # @cpu2(cache=True)
413
  def entry():
 
414
  os.chdir(os.path.abspath(os.path.dirname(__file__)))
415
  mp.set_start_method("spawn", force=True)
 
416
  pre[32,32]()
417
  ui[32,32]()
418
 
419
+ # entry
420
+
421
+ entry[32,32]()
422
+
423
  # end