mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-22 05:32:12 -03:00
11 lines
425 B
Python
11 lines
425 B
Python
from typing import Dict, Any
|
|
from .model_file_service import ProgressCallback
|
|
from .websocket_manager import ws_manager
|
|
|
|
|
|
class WebSocketProgressCallback(ProgressCallback):
|
|
"""WebSocket implementation of progress callback"""
|
|
|
|
async def on_progress(self, progress_data: Dict[str, Any]) -> None:
|
|
"""Send progress data via WebSocket"""
|
|
await ws_manager.broadcast_auto_organize_progress(progress_data) |