AIAnytime / AIAnytime/Phi-2-Fine-Tuning

Doubt on Training Model

未关闭
#1 0 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看
主要语言
Jupyter Notebook
星标
11
派生
7
PR 合并指标
30 天内没有已合并 PR

描述

I am trying to learn much about llms and finetuning , I watched your video i have a small doubt. After training process , we can save the new fine tuned model right? is it possible to use that saved model file to a streamlit app?

I tried the below code
```
trainer.train()
torch.save(trainer, 'saved_model.pth')
```
currently i am stuck on how to use it in a streamlit app.

贡献指南

这个仓库没有索引到贡献指南

评估

这个 Issue 还没有评估数据。

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。