Merge pull request #1529 from nullstreak/deepinfra-patch-1

Fix DeepInfra: Model is not supported
pull/1532/head 0.2.0.8
H Lohaus 8 months ago committed by GitHub
commit bf1d642ecf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -17,7 +17,8 @@ class DeepInfra(AsyncGeneratorProvider, ProviderModelMixin):
def get_models(cls):
if not cls.models:
url = 'https://api.deepinfra.com/models/featured'
cls.models = requests.get(url).json()
models = requests.get(url).json()
cls.models = [model['model_name'] for model in models]
return cls.models
@classmethod

Loading…
Cancel
Save