mirror of
https://github.com/jamiepine/voicebox.git
synced 2026-10-02 16:45:15 -07:00
Refactor provider health status and update build configurations
- Updated the `ProviderSettings` component to log the current active provider. - Changed the provider health status to use specific names for MLX and PyTorch backends. - Removed unnecessary exclusions from the build scripts for both PyTorch CPU and CUDA providers. - Ensured consistency in the `.spec` files for PyTorch providers by aligning exclusion lists.
This commit is contained in:
@@ -172,7 +172,8 @@ export function ProviderSettings() {
|
|||||||
const installedProviders = providersData?.installed || [];
|
const installedProviders = providersData?.installed || [];
|
||||||
|
|
||||||
// Determine current active provider
|
// Determine current active provider
|
||||||
const currentProvider = activeProvider?.provider || 'auto';
|
const currentProvider = activeProvider?.provider;
|
||||||
|
console.log('currentProvider', currentProvider);
|
||||||
const selectedProvider = currentProvider as ProviderType;
|
const selectedProvider = currentProvider as ProviderType;
|
||||||
|
|
||||||
return (
|
return (
|
||||||
|
|||||||
@@ -98,9 +98,12 @@ class BundledProvider:
|
|||||||
elif hasattr(backend, 'device'):
|
elif hasattr(backend, 'device'):
|
||||||
device = backend.device
|
device = backend.device
|
||||||
|
|
||||||
|
# Use apple-mlx for MLX backend, pytorch-cpu for PyTorch
|
||||||
|
provider_name = "apple-mlx" if backend_type == "mlx" else "pytorch-cpu"
|
||||||
|
|
||||||
return ProviderHealth(
|
return ProviderHealth(
|
||||||
status="healthy",
|
status="healthy",
|
||||||
provider=f"bundled-{backend_type}",
|
provider=provider_name,
|
||||||
version=None, # Provider versioning not implemented yet
|
version=None, # Provider versioning not implemented yet
|
||||||
model=model_size,
|
model=model_size,
|
||||||
device=device,
|
device=device,
|
||||||
|
|||||||
@@ -63,7 +63,6 @@ def build_provider():
|
|||||||
'--exclude-module', 'triton',
|
'--exclude-module', 'triton',
|
||||||
'--exclude-module', 'torch._dynamo',
|
'--exclude-module', 'torch._dynamo',
|
||||||
'--exclude-module', 'torch._inductor',
|
'--exclude-module', 'torch._inductor',
|
||||||
'--exclude-module', 'torch.testing',
|
|
||||||
'--exclude-module', 'torch.utils.benchmark',
|
'--exclude-module', 'torch.utils.benchmark',
|
||||||
'--exclude-module', 'IPython',
|
'--exclude-module', 'IPython',
|
||||||
'--exclude-module', 'matplotlib',
|
'--exclude-module', 'matplotlib',
|
||||||
|
|||||||
@@ -20,7 +20,7 @@ a = Analysis(
|
|||||||
hookspath=[],
|
hookspath=[],
|
||||||
hooksconfig={},
|
hooksconfig={},
|
||||||
runtime_hooks=[],
|
runtime_hooks=[],
|
||||||
excludes=['torch.utils.tensorboard', 'tensorboard', 'triton', 'torch._dynamo', 'torch._inductor', 'torch.testing', 'torch.utils.benchmark', 'IPython', 'matplotlib', 'PIL', 'cv2', 'torchvision', 'torchaudio'],
|
excludes=['torch.utils.tensorboard', 'tensorboard', 'triton', 'torch._dynamo', 'torch._inductor', 'torch.utils.benchmark', 'IPython', 'matplotlib', 'PIL', 'cv2', 'torchvision', 'torchaudio'],
|
||||||
noarchive=False,
|
noarchive=False,
|
||||||
optimize=0,
|
optimize=0,
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -65,7 +65,6 @@ def build_provider():
|
|||||||
'--exclude-module', 'triton',
|
'--exclude-module', 'triton',
|
||||||
'--exclude-module', 'torch._dynamo',
|
'--exclude-module', 'torch._dynamo',
|
||||||
'--exclude-module', 'torch._inductor',
|
'--exclude-module', 'torch._inductor',
|
||||||
'--exclude-module', 'torch.testing',
|
|
||||||
'--exclude-module', 'torch.utils.benchmark',
|
'--exclude-module', 'torch.utils.benchmark',
|
||||||
'--exclude-module', 'IPython',
|
'--exclude-module', 'IPython',
|
||||||
'--exclude-module', 'matplotlib',
|
'--exclude-module', 'matplotlib',
|
||||||
|
|||||||
@@ -20,7 +20,7 @@ a = Analysis(
|
|||||||
hookspath=[],
|
hookspath=[],
|
||||||
hooksconfig={},
|
hooksconfig={},
|
||||||
runtime_hooks=[],
|
runtime_hooks=[],
|
||||||
excludes=['torch.utils.tensorboard', 'tensorboard', 'triton', 'torch._dynamo', 'torch._inductor', 'torch.testing', 'torch.utils.benchmark', 'IPython', 'matplotlib', 'PIL', 'cv2', 'torchvision', 'torchaudio'],
|
excludes=['torch.utils.tensorboard', 'tensorboard', 'triton', 'torch._dynamo', 'torch._inductor', 'torch.utils.benchmark', 'IPython', 'matplotlib', 'PIL', 'cv2', 'torchvision', 'torchaudio'],
|
||||||
noarchive=False,
|
noarchive=False,
|
||||||
optimize=0,
|
optimize=0,
|
||||||
)
|
)
|
||||||
|
|||||||
Binary file not shown.
Reference in New Issue
Block a user