diff --git a/static/js/components/shared/ModelCard.js b/static/js/components/shared/ModelCard.js index 633e2825..d8b1cd5e 100644 --- a/static/js/components/shared/ModelCard.js +++ b/static/js/components/shared/ModelCard.js @@ -4,7 +4,7 @@ import { showModelModal } from './ModelModal.js'; import { toggleShowcase } from './showcase/ShowcaseView.js'; import { bulkManager } from '../../managers/BulkManager.js'; import { modalManager } from '../../managers/ModalManager.js'; -import { NSFW_LEVELS } from '../../utils/constants.js'; +import { NSFW_LEVELS, getBaseModelAbbreviation } from '../../utils/constants.js'; import { MODEL_TYPES } from '../../api/apiConfig.js'; import { getModelApiClient } from '../../api/modelApiFactory.js'; import { showDeleteModal } from '../../utils/modalUtils.js'; @@ -564,6 +564,9 @@ export function createModelCard(model, modelType) { ? 'fas fa-image' : 'fas fa-folder-open'; + const baseModelLabel = model.base_model || 'Unknown'; + const baseModelAbbreviation = getBaseModelAbbreviation(baseModelLabel); + card.innerHTML = `