nodes/kijai-ComfyUI-HunyuanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

compile fixes

Browse Source
This commit is contained in:
kijai
2024-12-23 11:42:20 +02:00
parent 1614fb2f3c
commit 2fe649be41
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
enhance_a_video/globals.py
+8 -3
View File
@@ -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