alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-21 21:22:11 -03:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add WebSocket support for checkpoint download progress and update related components

Browse Source
This commit is contained in:
Will Miao
2025-04-13 21:31:01 +08:00
parent 9822f2c614
commit 76fc9e5a3d
3 changed files with 95 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
static/js/managers/CheckpointDownloadManager.js
View File

@@ -292,9 +292,9 @@ export class CheckpointDownloadManager {
const updateProgress = this.loadingManager.showDownloadProgress(1);
updateProgress(0, 0, this.currentVersion.name);
// Setup WebSocket for progress updates
// Setup WebSocket for progress updates using checkpoint-specific endpoint
const wsProtocol = window.location.protocol === 'https:' ? 'wss://' : 'ws://';
const ws = new WebSocket(`${wsProtocol}${window.location.host}/ws/fetch-progress`);
const ws = new WebSocket(`${wsProtocol}${window.location.host}/ws/checkpoint-progress`);
ws.onmessage = (event) => {
const data = JSON.parse(event.data);