YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

GPT-2 Instruct Model

This is a fine-tuned GPT-2 model for instruction-following tasks.

Base Model

  • openai-community/gpt2

Training Details

  • Epochs: 4
  • Batch size: 2
  • Max length: 512
  • Optimizer: AdamW

Dataset

  • Custom instruction dataset

Usage Example

from transformers import pipeline

chat = pipeline( "text-generation", model="Sanjarbek1024/gpt2-instruct", tokenizer="Sanjarbek1024/gpt2-instruct" )

prompt = "<|user|>\nWhat is AI?\n<|assistant|>\n"

result = chat(prompt, max_new_tokens=100) print(result[0]["generated_text"])

Author

  • Sanjarbek
Downloads last month
6
Safetensors
Model size
0.1B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Sanjarbek1024/gpt2-instruct

Quantizations
1 model