forked from KoboldAI/KoboldAI-Client
-
Notifications
You must be signed in to change notification settings - Fork 147
Open
Description
Hi I discover aihorde does allown you set model even you have choice one, it use random one in current code
KoboldAI/modeling/inference_models/horde/class.py
def _raw_generate(
we can found this code
cluster_metadata = {
"prompt": decoded_prompt,
"params": reqdata,
"models": [x for x in utils.koboldai_vars.cluster_requested_models if x],
"trusted_workers": False,
}
the var utils.koboldai_vars.cluster_requested_models is never set.
so I did change the code to this
matched_models = [x for x in self.models if x.get('name') == self.model_name]
cluster_metadata = {
"prompt": decoded_prompt,
"params": reqdata,
"models": self.model,
"trusted_workers": False,
}
now it takes the model I choice, but I think it should find all model that are simulre with different names and do a array of it so it use more server
Metadata
Metadata
Assignees
Labels
No labels