compile fixes

kijai · Dec 23, 2024 · 2fe649b · 2fe649b
1 parent 1614fb2
commit 2fe649b
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/enhance_a_video/globals.py b/enhance_a_video/globals.py
@@ -1,14 +1,16 @@
+import torch
+
 NUM_FRAMES = None
 FETA_WEIGHT = None
 ENABLE_FETA_SINGLE = False
 ENABLE_FETA_DOUBLE = False
 
-
+@torch.compiler.disable()
 def set_num_frames(num_frames: int):
     global NUM_FRAMES
     NUM_FRAMES = num_frames
 
-
+@torch.compiler.disable()
 def get_num_frames() -> int:
     return NUM_FRAMES
 
@@ -23,16 +25,19 @@ def disable_enhance():
     ENABLE_FETA_SINGLE = False
     ENABLE_FETA_DOUBLE = False
 
+@torch.compiler.disable()
 def is_enhance_enabled_single() -> bool:
     return ENABLE_FETA_SINGLE
 
+@torch.compiler.disable()
 def is_enhance_enabled_double() -> bool:
     return ENABLE_FETA_DOUBLE
 
+@torch.compiler.disable()
 def set_enhance_weight(feta_weight: float):
     global FETA_WEIGHT
     FETA_WEIGHT = feta_weight
 
-
+@torch.compiler.disable()
 def get_enhance_weight() -> float:
     return FETA_WEIGHT