NickyNicky's picture
Update README.md
fe00e51 verified
metadata
library_name: transformers
license: apache-2.0
datasets:
  - NickyNicky/oasst2_clusters
  - OpenAssistant/oasst2
model:
  - google/gemma-1.1-2b-it
language:
  - bg
  - ca
  - cs
  - da
  - de
  - en
  - es
  - fr
  - hr
  - hu
  - it
  - nl
  - pl
  - pt
  - ro
  - ru
  - sl
  - sr
  - sv
  - uk
widget:
  - text: |
      <bos><start_of_turn>system
      You are a helpful AI assistant.<end_of_turn>
      <start_of_turn>user
      {question}<end_of_turn>
      <start_of_turn>model

Metrics.

TrainOutput(global_step=1390,
  training_loss=1.0502444919064748,
  metrics={
    'train_runtime': 22700.8355,
    'train_samples_per_second': 2.449,
    'train_steps_per_second': 0.061,
    'total_flos': 1.2395973405265306e+18,
    'train_loss': 1.0502444919064748,
    'epoch': 4.05
})

Take dataset.

OpenAssistant/oasst2

Dataset format gemma fine tune.

NickyNicky/oasst2_clusters

colab examples.

https://colab.research.google.com/drive/16qS7NMSu20LzcwvYCrBGVI7rd9Hr-vpN?usp=sharing