Spaces:
Sleeping
Sleeping
johannoriel
commited on
Commit
•
dbc6bab
1
Parent(s):
89c9051
Update plugins/ragllm.py
Browse fileschange default to huggingface
- plugins/ragllm.py +2 -2
plugins/ragllm.py
CHANGED
@@ -163,7 +163,7 @@ class RagllmPlugin(Plugin):
|
|
163 |
updated_config = {}
|
164 |
|
165 |
# Load config file only if provider is not huggingface
|
166 |
-
current_provider = config.get('provider', '
|
167 |
if current_provider != 'huggingface':
|
168 |
self.config = self.load_llm_config()
|
169 |
|
@@ -220,7 +220,7 @@ class RagllmPlugin(Plugin):
|
|
220 |
return updated_config
|
221 |
|
222 |
def get_sidebar_config_ui(self, config: Dict[str, Any]) -> Dict[str, Any]:
|
223 |
-
provider = config.get('provider', '
|
224 |
available_models = self.get_available_models(provider)
|
225 |
default_model = config.get('llm_model', available_models[0] if available_models else None)
|
226 |
|
|
|
163 |
updated_config = {}
|
164 |
|
165 |
# Load config file only if provider is not huggingface
|
166 |
+
current_provider = config.get('provider', 'huggingface')
|
167 |
if current_provider != 'huggingface':
|
168 |
self.config = self.load_llm_config()
|
169 |
|
|
|
220 |
return updated_config
|
221 |
|
222 |
def get_sidebar_config_ui(self, config: Dict[str, Any]) -> Dict[str, Any]:
|
223 |
+
provider = config.get('provider', 'huggingface')
|
224 |
available_models = self.get_available_models(provider)
|
225 |
default_model = config.get('llm_model', available_models[0] if available_models else None)
|
226 |
|