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

feat: Refactor model type determination to use state for saving metadata and handling events

Browse Source
This commit is contained in:
Will Miao
2025-07-22 16:44:21 +08:00
parent fcfc868e57
commit c82fabb67f
4 changed files with 8 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
static/js/components/shared/ModelMetadata.js
View File

@@ -114,9 +114,7 @@ export function setupModelNameEditing(filePath) {
// Get the file path from the dataset
const filePath = this.dataset.filePath;
// Determine model type based on file extension
const isCheckpoint = filePath.includes('.safetensors') || filePath.includes('.ckpt');
const saveFunction = isCheckpoint ? saveCheckpointMetadata : saveLoraMetadata;
const saveFunction = state.currentPageType === 'checkpoints' ? saveCheckpointMetadata : saveLoraMetadata;
await saveFunction(filePath, { model_name: newModelName });
@@ -297,9 +295,7 @@ async function saveBaseModel(filePath, originalValue) {
}
try {
// Determine model type based on file extension
const isCheckpoint = filePath.includes('.safetensors') || filePath.includes('.ckpt');
const saveFunction = isCheckpoint ? saveCheckpointMetadata : saveLoraMetadata;
const saveFunction = state.currentPageType === 'checkpoints' ? saveCheckpointMetadata : saveLoraMetadata;
await saveFunction(filePath, { base_model: newBaseModel });
@@ -421,19 +417,10 @@ export function setupFileNameEditing(filePath) {
// Get the file path from the dataset
const filePath = this.dataset.filePath;
// Determine model type and use appropriate rename function
const isCheckpoint = filePath.includes('.safetensors') || filePath.includes('.ckpt');
let result;
if (isCheckpoint) {
// Use checkpoint rename function if it exists, otherwise fallback to generic approach
if (typeof renameCheckpointFile === 'function') {
result = await renameCheckpointFile(filePath, newFileName);
} else {
// Fallback: use checkpoint metadata save function
await saveCheckpointMetadata(filePath, { file_name: newFileName });
result = { success: true };
}
if (state.currentPageType === 'checkpoints') {
result = await renameCheckpointFile(filePath, newFileName);
} else {
// Use LoRA rename function
result = await renameLoraFile(filePath, newFileName);