vnyash
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
from typing import Optional
|
|
import gradio
|
|
|
|
import deepfuze.globals
|
|
import deepfuze.choices
|
|
from deepfuze.typing import VideoMemoryStrategy
|
|
from deepfuze import wording
|
|
|
|
VIDEO_MEMORY_STRATEGY_DROPDOWN : Optional[gradio.Dropdown] = None
|
|
SYSTEM_MEMORY_LIMIT_SLIDER : Optional[gradio.Slider] = None
|
|
|
|
|
|
def render() -> None:
|
|
global VIDEO_MEMORY_STRATEGY_DROPDOWN
|
|
global SYSTEM_MEMORY_LIMIT_SLIDER
|
|
|
|
VIDEO_MEMORY_STRATEGY_DROPDOWN = gradio.Dropdown(
|
|
label = wording.get('uis.video_memory_strategy_dropdown'),
|
|
choices = deepfuze.choices.video_memory_strategies,
|
|
value = deepfuze.globals.video_memory_strategy
|
|
)
|
|
SYSTEM_MEMORY_LIMIT_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.system_memory_limit_slider'),
|
|
step =deepfuze.choices.system_memory_limit_range[1] - deepfuze.choices.system_memory_limit_range[0],
|
|
minimum = deepfuze.choices.system_memory_limit_range[0],
|
|
maximum = deepfuze.choices.system_memory_limit_range[-1],
|
|
value = deepfuze.globals.system_memory_limit
|
|
)
|
|
|
|
|
|
def listen() -> None:
|
|
VIDEO_MEMORY_STRATEGY_DROPDOWN.change(update_video_memory_strategy, inputs = VIDEO_MEMORY_STRATEGY_DROPDOWN)
|
|
SYSTEM_MEMORY_LIMIT_SLIDER.release(update_system_memory_limit, inputs = SYSTEM_MEMORY_LIMIT_SLIDER)
|
|
|
|
|
|
def update_video_memory_strategy(video_memory_strategy : VideoMemoryStrategy) -> None:
|
|
deepfuze.globals.video_memory_strategy = video_memory_strategy
|
|
|
|
|
|
def update_system_memory_limit(system_memory_limit : int) -> None:
|
|
deepfuze.globals.system_memory_limit = system_memory_limit
|