New architecture

This commit is contained in:
Niels Geens
2024-10-31 18:08:35 +01:00
parent bd5c5911af
commit ee089e5be7
6 changed files with 347 additions and 140 deletions

View File

@@ -75,14 +75,4 @@ class LlmEngine:
)
thread = Thread(target=self.pipeline, kwargs=pipeline_kwargs)
thread.start()
return util.stop_before_value(streamer, '<|eot_id|>')
def finetune(self, dataset_paths: list, output_dir: str):
"""
Fine-tune the model with new datasets and save the updated model weights.
Args:
dataset_paths: List of paths to datasets for fine-tuning.
output_dir: Directory where the updated model weights will be saved.
"""
pass
return util.stop_before_value(streamer, '<|eot_id|>')