deepfuze

2024-06-13 07:56:13 +05:30
commit 47d3520c19
184 changed files with 10075 additions and 0 deletions
@@ -0,0 +1,41 @@
+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