diff --git a/video_upscale_model.py b/video_upscale_model.py index 804bd25..67bd32a 100644 --- a/video_upscale_model.py +++ b/video_upscale_model.py @@ -8,6 +8,14 @@ from PIL import Image import gc import time +# Add the missing progress bar initialization function +def init_comfyui_progress(): + """ + Initialize ComfyUI progress bar system for this module. + This function is called by ComfyUI's progress bar system. + """ + pass # No special initialization needed for our progress implementation + class Video_Upscale_With_Model: """ A memory-efficient implementation for upscaling video frames using an upscale model @@ -348,4 +356,4 @@ NODE_CLASS_MAPPINGS = { NODE_DISPLAY_NAME_MAPPINGS = { "Video_Upscale_With_Model": "Video Upscale With Model", "Free_Video_Memory": "Free Video Memory", -} \ No newline at end of file +}