diff --git a/web/containers/ModelDownloadButton/index.tsx b/web/containers/ModelDownloadButton/index.tsx index 946c2b2f8..9555582fa 100644 --- a/web/containers/ModelDownloadButton/index.tsx +++ b/web/containers/ModelDownloadButton/index.tsx @@ -3,6 +3,8 @@ import { useCallback, useMemo } from 'react' import { Button, Tooltip } from '@janhq/joi' import { useAtomValue, useSetAtom } from 'jotai' +import { twMerge } from 'tailwind-merge' + import { MainViewState } from '@/constants/screens' import { useCreateNewThread } from '@/hooks/useCreateNewThread' @@ -22,8 +24,9 @@ interface Props { id: string theme?: 'primary' | 'ghost' | 'icon' | 'destructive' | undefined variant?: 'solid' | 'soft' | 'outline' | undefined + className?: string } -const ModelDownloadButton = ({ id, theme, variant }: Props) => { +const ModelDownloadButton = ({ id, theme, variant, className }: Props) => { const { downloadModel } = useDownloadModel() const downloadingModels = useAtomValue(getDownloadingModelAtom) const downloadedModels = useAtomValue(downloadedModelsAtom) @@ -59,7 +62,8 @@ const ModelDownloadButton = ({ id, theme, variant }: Props) => { const defaultButton = (