library_name: transformers
tags:
- llm
- 7b
license: cc-by-4.0
language:
- en
datasets:
- argilla/distilabel-math-preference-dpo
Jaskier-7b-dpo-v5.6
This is work-in-progress model, may not be ready for production use
Model based on paulml/OGNO-7B (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:
from transformers import pipeline, Conversation
import torch
base_model_name = "bardsai/jaskier-7b-dpo-v5.6"
chatbot = pipeline("conversational", model=base_model_name, torch_dtype=torch.float16, device_map="auto")
conversation = Conversation("Can Poland run into space?")
conversation = chatbot(conversation)
print(conversation.messages[-1]["content"])
Output
"Poland, as a nation, doesn't physically run or possess the capability to physically travel into space. However, Poland, like many other countries, supports scientific research and space exploration through its institutions and collaborations with international space agencies."
If you still find any issues with "INST" character chain appearing in generated output, try our newest model: https://huggingface.co/bardsai/jaskier-7b-dpo-v6.1 . Re-tasking the prompt can also help.
Changelog
- 2024-02-16: 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 info@bards.ai