Skip to content

aihorde do not set model #526

@greatlord

Description

@greatlord

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

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions