ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-24 06:46:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

enhanced mix it up

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-07-11 01:27:29 +02:00
parent d29bde9d0a
commit aa21fdfb49
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lollms/personality.py
View File

@ -3769,8 +3769,10 @@ class APScript(StateMachine):
'initial_prompt': prompt, 'initial_prompt': prompt,
'rounds': [] 'rounds': []
} }
for rounds in range(nb_rounds): for round in range(nb_rounds):
self.step_start(f"Round {round + 1}")
for idx, model_id in enumerate(models): for idx, model_id in enumerate(models):
self.step_start(f"Using model {model_id}")
binding_name, model_name = model_id.split("::") binding_name, model_name = model_id.split("::")
self.select_model(binding_name, model_name) self.select_model(binding_name, model_name)
@ -3785,6 +3787,8 @@ class APScript(StateMachine):
'output': output 'output': output
}) })
previous_outputs.append((model_id, output)) # Update for the next round previous_outputs.append((model_id, output)) # Update for the next round
self.step_end(f"Using model {model_id}")
self.step_end(f"Round {round + 1}")
# Final round with the master model # Final round with the master model
self.select_model(*master_model.split("::")) self.select_model(*master_model.split("::"))