Spaces:
Paused
Paused
File size: 2,406 Bytes
1030ba2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
# ui/shared_state.py
import threading
# --- Application-wide Threading Controls ---
# Lock for thread-safe queue modifications.
queue_lock = threading.Lock()
# Event to signal the abortion of the current processing task.
abort_event = threading.Event()
# --- Model and Global State Containers ---
# This dictionary will be populated at runtime by the main script after the models are loaded.
# It allows other modules to access the models without circular imports.
models = {}
# This dictionary holds the application state, which is passed to the atexit
# handler to enable the autosave functionality on browser close or exit.
global_state_for_autosave = {}
# --- UI and Parameter Mapping Constants ---
# These constants define the structure of the UI and how UI components
# map to the parameters of the backend generation worker.
# Creative "Recipe" Parameters (for portable PNG metadata and task editing)
CREATIVE_PARAM_KEYS = [
'prompt', 'n_prompt', 'total_second_length', 'seed', 'preview_frequency_ui',
'segments_to_decode_csv', 'gs_ui', 'gs_schedule_shape_ui', 'gs_final_ui', 'steps', 'cfg', 'rs'
]
# Environment/Debug Parameters (for the full workspace, machine/session-specific)
ENVIRONMENT_PARAM_KEYS = [
'use_teacache', 'use_fp32_transformer_output_ui', 'gpu_memory_preservation',
'mp4_crf', 'output_folder_ui', 'latent_window_size'
]
# A comprehensive list of all UI components that define a task's parameters.
ALL_TASK_UI_KEYS = CREATIVE_PARAM_KEYS + ENVIRONMENT_PARAM_KEYS
# This maps the string keys of the Gradio UI components to the keyword argument
# names expected by the 'worker' function in generation_core.py.
UI_TO_WORKER_PARAM_MAP = {
'prompt': 'prompt',
'n_prompt': 'n_prompt',
'total_second_length': 'total_second_length',
'seed': 'seed',
'use_teacache': 'use_teacache',
'preview_frequency_ui': 'preview_frequency',
'segments_to_decode_csv': 'segments_to_decode_csv',
'gs_ui': 'gs',
'gs_schedule_shape_ui': 'gs_schedule_active',
'gs_final_ui': 'gs_final',
'steps': 'steps',
'cfg': 'cfg',
'latent_window_size': 'latent_window_size',
'gpu_memory_preservation': 'gpu_memory_preservation',
'use_fp32_transformer_output_ui': 'use_fp32_transformer_output',
'rs': 'rs',
'mp4_crf': 'mp4_crf',
'output_folder_ui': 'output_folder'
}
|