mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-26 07:35:44 -03:00
feat(initialization): add support for embeddings page type and log progress updates
This commit is contained in:
@@ -56,6 +56,8 @@ class InitializationManager {
|
|||||||
this.pageType = 'checkpoints';
|
this.pageType = 'checkpoints';
|
||||||
} else if (path.includes('/loras')) {
|
} else if (path.includes('/loras')) {
|
||||||
this.pageType = 'loras';
|
this.pageType = 'loras';
|
||||||
|
} else if (path.includes('/embeddings')) {
|
||||||
|
this.pageType = 'embeddings';
|
||||||
} else {
|
} else {
|
||||||
// Default to loras if can't determine
|
// Default to loras if can't determine
|
||||||
this.pageType = 'loras';
|
this.pageType = 'loras';
|
||||||
@@ -195,6 +197,7 @@ class InitializationManager {
|
|||||||
*/
|
*/
|
||||||
handleProgressUpdate(data) {
|
handleProgressUpdate(data) {
|
||||||
if (!data) return;
|
if (!data) return;
|
||||||
|
console.log('Received progress update:', data);
|
||||||
|
|
||||||
// Check if this update is for our page type
|
// Check if this update is for our page type
|
||||||
if (data.pageType && data.pageType !== this.pageType) {
|
if (data.pageType && data.pageType !== this.pageType) {
|
||||||
@@ -206,7 +209,8 @@ class InitializationManager {
|
|||||||
if (!data.pageType && data.scanner_type) {
|
if (!data.pageType && data.scanner_type) {
|
||||||
const scannerTypeToPageType = {
|
const scannerTypeToPageType = {
|
||||||
'lora': 'loras',
|
'lora': 'loras',
|
||||||
'checkpoint': 'checkpoints'
|
'checkpoint': 'checkpoints',
|
||||||
|
'embedding': 'embeddings'
|
||||||
};
|
};
|
||||||
|
|
||||||
if (scannerTypeToPageType[data.scanner_type] !== this.pageType) {
|
if (scannerTypeToPageType[data.scanner_type] !== this.pageType) {
|
||||||
|
|||||||
Reference in New Issue
Block a user