|
--- |
|
library_name: transformers |
|
tags: |
|
- llm |
|
- 7b |
|
license: cc-by-4.0 |
|
datasets: |
|
- jondurbin/truthy-dpo-v0.1 |
|
language: |
|
- en |
|
--- |
|
|
|
# Jaskier-7b-dpo-v5.6 |
|
|
|
**This is work-in-progress model, may not be ready for production use** |
|
|
|
Model based on `bardsai/jaskier-7b-dpo-v5.6` (downstream version of Mistral7B) finetuned using Direct Preference Optimization on argilla/distilabel-math-preference-dpo. |
|
|
|
## How to use |
|
|
|
You can use this model directly with a Hugging Face pipeline: |
|
```python |
|
|
|
from transformers import pipeline, Conversation |
|
import torch |
|
|
|
base_model_name = "bardsai/jaskier-7b-dpo-v6.1" |
|
chatbot = pipeline("conversational", model=base_model_name, torch_dtype=torch.float16, device_map="auto") |
|
conversation = Conversation("Can Poland into space?") |
|
conversation = chatbot(conversation) |
|
print(conversation.messages[-1]["content"]) |
|
|
|
``` |
|
|
|
## Output |
|
|
|
"Poland, as a nation, doesn't physically travel to space. However, Poland has contributed to the field of space exploration through its scientists, engineers, and collaborations with international space agencies. The Polish Space Agency, established in 2016, aims to promote and coordinate the country's space activities." |
|
|
|
## Changelog |
|
|
|
- 2024-02-20: Initial release |
|
|
|
## About bards.ai |
|
|
|
At bards.ai, we focus on providing machine learning expertise and skills to our partners, particularly in the areas of nlp, machine vision and time series analysis. Our team is located in Wroclaw, Poland. Please visit our website for more information: bards.ai |
|
|
|
Let us know if you use our model :). Also, if you need any help, feel free to contact us at [email protected] |
