This is a LoRA finetuned model of gemma-2B for crewai library that produces the Goal and Backstory description automatically in it’s agent method or agent() by taking only Role as user input, this helps to generate proper descriptions of those parameters through a llm instead of manually writing it. We have 103 rows of descriptions of different roles and their respective goals and backstory which is used to train the models, see this dataset for details. The model gives a training loss of 0.931 on training data with 25 setps batch size as shown in below figure: As the Model is trained on tiny dataset it may yeild inaccurate output sometimes, to avoid this we can train the model on a bigger dataset in future.
Модальности:
Генерация текста
Задача: Генерация текста
Автор: DrDrek
Теги: gemma, en, text-generation-inference, endpoints_compatible
Лайков: 3 | Загрузок: 40
Описание основано на материалах HuggingFace. Перевод выполнен автоматически.
