@@ -2,6 +2,7 @@ import { Flex } from '@invoke-ai/ui-library';
22import { useAppSelector } from 'app/store/storeHooks' ;
33import ScrollableContent from 'common/components/OverlayScrollbars/ScrollableContent' ;
44import { memo , useMemo } from 'react' ;
5+ import { useTranslation } from 'react-i18next' ;
56import {
67 useControlNetModels ,
78 useEmbeddingModels ,
@@ -18,6 +19,7 @@ import { ModelListWrapper } from './ModelListWrapper';
1819
1920const ModelList = ( ) => {
2021 const { searchTerm, filteredModelType } = useAppSelector ( ( s ) => s . modelmanagerV2 ) ;
22+ const { t } = useTranslation ( ) ;
2123
2224 const [ mainModels , { isLoading : isLoadingMainModels } ] = useMainModels ( ) ;
2325 const filteredMainModels = useMemo (
@@ -67,18 +69,22 @@ const ModelList = () => {
6769 { /* Main Model List */ }
6870 { isLoadingMainModels && < FetchingModelsLoader loadingMessage = "Loading Main Models..." /> }
6971 { ! isLoadingMainModels && filteredMainModels . length > 0 && (
70- < ModelListWrapper title = "Main" modelList = { filteredMainModels } key = "main" />
72+ < ModelListWrapper title = { t ( 'modelManager.main' ) } modelList = { filteredMainModels } key = "main" />
7173 ) }
7274 { /* LoRAs List */ }
7375 { isLoadingLoRAModels && < FetchingModelsLoader loadingMessage = "Loading LoRAs..." /> }
7476 { ! isLoadingLoRAModels && filteredLoRAModels . length > 0 && (
75- < ModelListWrapper title = "LoRA" modelList = { filteredLoRAModels } key = "loras" />
77+ < ModelListWrapper title = { t ( 'modelManager.loraModels' ) } modelList = { filteredLoRAModels } key = "loras" />
7678 ) }
7779
7880 { /* TI List */ }
79- { isLoadingEmbeddingModels && < FetchingModelsLoader loadingMessage = "Loading Embeddings ..." /> }
81+ { isLoadingEmbeddingModels && < FetchingModelsLoader loadingMessage = "Loading Textual Inversions ..." /> }
8082 { ! isLoadingEmbeddingModels && filteredEmbeddingModels . length > 0 && (
81- < ModelListWrapper title = "Embedding" modelList = { filteredEmbeddingModels } key = "textual-inversions" />
83+ < ModelListWrapper
84+ title = { t ( 'modelManager.textualInversions' ) }
85+ modelList = { filteredEmbeddingModels }
86+ key = "textual-inversions"
87+ />
8288 ) }
8389
8490 { /* VAE List */ }
@@ -95,12 +101,16 @@ const ModelList = () => {
95101 { /* IP Adapter List */ }
96102 { isLoadingIPAdapterModels && < FetchingModelsLoader loadingMessage = "Loading IP Adapters..." /> }
97103 { ! isLoadingIPAdapterModels && filteredIPAdapterModels . length > 0 && (
98- < ModelListWrapper title = "IP Adapter" modelList = { filteredIPAdapterModels } key = "ip-adapters" />
104+ < ModelListWrapper
105+ title = { t ( 'modelManager.ipAdapters' ) }
106+ modelList = { filteredIPAdapterModels }
107+ key = "ip-adapters"
108+ />
99109 ) }
100110 { /* T2I Adapters List */ }
101111 { isLoadingT2IAdapterModels && < FetchingModelsLoader loadingMessage = "Loading T2I Adapters..." /> }
102112 { ! isLoadingT2IAdapterModels && filteredT2IAdapterModels . length > 0 && (
103- < ModelListWrapper title = "T2I Adapter" modelList = { filteredT2IAdapterModels } key = "t2i-adapters" />
113+ < ModelListWrapper title = { t ( 'common.t2iAdapter' ) } modelList = { filteredT2IAdapterModels } key = "t2i-adapters" />
104114 ) }
105115 </ Flex >
106116 </ ScrollableContent >
0 commit comments