Text Generation
Transformers
Safetensors
English
gpt2
conversational
gpt
chatbot
finetune
text-generation-inference
My DialoGPT Model
This is a fine-tuned version of microsoft/DialoGPT-small on custom data about Dominica.
Model Details
- Model Name: DialoGPT-small
- Training Data: Custom dataset about Dominica
- Evaluation: Achieved
eval_lossof 12.85
Usage
To use this model, you can load it as follows:
from transformers import AutoModelForCausalLM, AutoTokenizer
# Load the model and tokenizer
model_name = "unknownCode/IslandBoyRepo"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)
# Generate a response
input_text = "What is the capital of Dominica?"
inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**inputs)
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(response)
- Downloads last month
- 7
Install from pip and serve model
# Install vLLM from pip: pip install vllm# Start the vLLM server: vllm serve "unknownCode/IslandBoyRepo"# Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "unknownCode/IslandBoyRepo", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'