upgraded ui

This commit is contained in:
ParisNeo 2023-05-11 22:45:50 +02:00
parent 8fbcdfbe12
commit dfd2c2f6c2
2 changed files with 5 additions and 4 deletions

View File

@ -17,12 +17,12 @@ __github__ = "https://github.com/nomic-ai/gpt4all-ui"
__copyright__ = "Copyright 2023, " __copyright__ = "Copyright 2023, "
__license__ = "Apache 2.0" __license__ = "Apache 2.0"
backend_name = "LLAMACPP" backend_name = "GPT4ALL"
class LLAMACPP(GPTBackend): class GPT4ALL(GPTBackend):
file_extension='*.bin' file_extension='*.bin'
def __init__(self, config:dict) -> None: def __init__(self, config:dict) -> None:
"""Builds a LLAMACPP backend """Builds a GPT4ALL backend
Args: Args:
config (dict): The configuration file config (dict): The configuration file
@ -30,7 +30,7 @@ class LLAMACPP(GPTBackend):
super().__init__(config, False) super().__init__(config, False)
self.model = Model( self.model = Model(
model_path=f"./models/llama_cpp/{self.config['model']}", model_path=f"./models/gpt4all/{self.config['model']}",
prompt_context="", prompt_prefix="", prompt_suffix="", prompt_context="", prompt_prefix="", prompt_suffix="",
n_ctx=self.config['ctx_size'], n_ctx=self.config['ctx_size'],
seed=self.config['seed'], seed=self.config['seed'],

View File

@ -347,6 +347,7 @@ export default {
}); });
}, },
onSelected(model_object){ onSelected(model_object){
console.log("Selected model")
update_setting('model', model_object.title) update_setting('model', model_object.title)
}, },
// Model installation // Model installation