Edit model card

v3

This model is a fine-tuned version of TheBloke/Mistral-7B-Instruct-v0.2-GPTQ on the hieunguyenminh/roleplay dataset.

Model description

This model can adapt to any type of characters and provide answer that personalize that character.

Training and evaluation data

It is trained with supervised learning and will be trained with DPO in the future.

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 0.0002
  • train_batch_size: 16
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: cosine
  • training_steps: 400
  • mixed_precision_training: Native AMP

Training results

Loss after 400 steps: 0.73

Framework versions

  • Transformers 4.35.2
  • Pytorch 2.1.0+cu121
  • Datasets 2.16.1
  • Tokenizers 0.15.0
Downloads last month
6
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for hieunguyenminh/v3

Dataset used to train hieunguyenminh/v3

Collection including hieunguyenminh/v3