fix: search hgf repo and downloaded filter
This commit is contained in:
parent
2801fde181
commit
5f9f766965
@ -205,7 +205,8 @@ function Hub() {
|
|||||||
|
|
||||||
if (
|
if (
|
||||||
e.target.value.length &&
|
e.target.value.length &&
|
||||||
(e.target.value.includes('/') || e.target.value.startsWith('http'))
|
(e.target.value.includes('/') || e.target.value.startsWith('http')) &&
|
||||||
|
!showOnlyDownloaded
|
||||||
) {
|
) {
|
||||||
setIsSearching(true)
|
setIsSearching(true)
|
||||||
|
|
||||||
@ -222,7 +223,8 @@ function Hub() {
|
|||||||
!sources.some(
|
!sources.some(
|
||||||
(s) =>
|
(s) =>
|
||||||
catalogModel.model_name.trim().split('/').pop() ===
|
catalogModel.model_name.trim().split('/').pop() ===
|
||||||
s.model_name.trim()
|
s.model_name.trim() &&
|
||||||
|
catalogModel.developer.trim() === s.developer?.trim()
|
||||||
)
|
)
|
||||||
) {
|
) {
|
||||||
setHuggingFaceRepo(catalogModel)
|
setHuggingFaceRepo(catalogModel)
|
||||||
@ -508,7 +510,46 @@ function Hub() {
|
|||||||
<div className="flex items-center gap-2">
|
<div className="flex items-center gap-2">
|
||||||
<Switch
|
<Switch
|
||||||
checked={showOnlyDownloaded}
|
checked={showOnlyDownloaded}
|
||||||
onCheckedChange={setShowOnlyDownloaded}
|
onCheckedChange={(checked) => {
|
||||||
|
setShowOnlyDownloaded(checked)
|
||||||
|
if (checked) {
|
||||||
|
setHuggingFaceRepo(null)
|
||||||
|
} else if (
|
||||||
|
searchValue.length &&
|
||||||
|
(searchValue.includes('/') || searchValue.startsWith('http'))
|
||||||
|
) {
|
||||||
|
// Re-trigger HuggingFace search when switching back to "All models"
|
||||||
|
setIsSearching(true)
|
||||||
|
if (addModelSourceTimeoutRef.current) {
|
||||||
|
clearTimeout(addModelSourceTimeoutRef.current)
|
||||||
|
}
|
||||||
|
addModelSourceTimeoutRef.current = setTimeout(async () => {
|
||||||
|
try {
|
||||||
|
const repoInfo = await fetchHuggingFaceRepo(
|
||||||
|
searchValue,
|
||||||
|
huggingfaceToken
|
||||||
|
)
|
||||||
|
if (repoInfo) {
|
||||||
|
const catalogModel = convertHfRepoToCatalogModel(repoInfo)
|
||||||
|
if (
|
||||||
|
!sources.some(
|
||||||
|
(s) =>
|
||||||
|
catalogModel.model_name.trim().split('/').pop() ===
|
||||||
|
s.model_name.trim() &&
|
||||||
|
catalogModel.developer.trim() === s.developer?.trim()
|
||||||
|
)
|
||||||
|
) {
|
||||||
|
setHuggingFaceRepo(catalogModel)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Error fetching repository info:', error)
|
||||||
|
} finally {
|
||||||
|
setIsSearching(false)
|
||||||
|
}
|
||||||
|
}, 500)
|
||||||
|
}
|
||||||
|
}}
|
||||||
/>
|
/>
|
||||||
<span className="text-xs text-main-view-fg/70 font-medium whitespace-nowrap">
|
<span className="text-xs text-main-view-fg/70 font-medium whitespace-nowrap">
|
||||||
{t('hub:downloaded')}
|
{t('hub:downloaded')}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user