44 lines
990 B
Python
44 lines
990 B
Python
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
|
|
|
|
|
|
def enable_enhance(single, double):
|
|
global ENABLE_FETA_SINGLE, ENABLE_FETA_DOUBLE
|
|
ENABLE_FETA_SINGLE = single
|
|
ENABLE_FETA_DOUBLE = double
|
|
|
|
def disable_enhance():
|
|
global ENABLE_FETA_SINGLE, ENABLE_FETA_DOUBLE
|
|
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
|