Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
improve get available model provider standalone test
Browse files
backend/tasks/get_available_model_provider.py
CHANGED
@@ -126,7 +126,7 @@ def get_available_model_provider(model_name, verbose=False):
|
|
126 |
if verbose:
|
127 |
logger.error(f"Error in get_available_model_provider: {str(e)}")
|
128 |
return None
|
129 |
-
|
130 |
if __name__ == "__main__":
|
131 |
|
132 |
models = [
|
@@ -143,7 +143,9 @@ if __name__ == "__main__":
|
|
143 |
|
144 |
for model in models:
|
145 |
provider = get_available_model_provider(model, verbose=True)
|
146 |
-
providers.append(provider)
|
|
|
|
|
|
|
147 |
|
148 |
-
print(f"Providers {len(providers)}: {providers}")
|
149 |
-
|
|
|
126 |
if verbose:
|
127 |
logger.error(f"Error in get_available_model_provider: {str(e)}")
|
128 |
return None
|
129 |
+
|
130 |
if __name__ == "__main__":
|
131 |
|
132 |
models = [
|
|
|
143 |
|
144 |
for model in models:
|
145 |
provider = get_available_model_provider(model, verbose=True)
|
146 |
+
providers.append((model, provider))
|
147 |
+
|
148 |
+
for model, provider in providers:
|
149 |
+
print(f"Model: {model}, Provider: {provider}")
|
150 |
|
151 |
+
print(f"Total Providers {len(providers)}: {providers}")
|
|