Karols commited on
Commit
15ea92b
1 Parent(s): 25c0f5c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +42 -0
README.md CHANGED
@@ -6,13 +6,55 @@ tags:
6
  license: cc-by-4.0
7
  language:
8
  - en
 
 
9
  ---
10
 
11
  # Jaskier-7b-dpo-v5.6
12
 
13
  **This is work-in-progress model, may not be ready for production use**
14
 
 
15
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
  ## Changelog
17
 
18
  - 2024-02-16: Initial release
 
6
  license: cc-by-4.0
7
  language:
8
  - en
9
+ datasets:
10
+ - argilla/distilabel-math-preference-dpo
11
  ---
12
 
13
  # Jaskier-7b-dpo-v5.6
14
 
15
  **This is work-in-progress model, may not be ready for production use**
16
 
17
+ Model based on `paulml/OGNO-7B` (downstream version of Mistral7B) finetuned using Direct Preference Optimization on argilla/distilabel-math-preference-dpo.
18
 
19
+ ## How to use
20
+
21
+ You can use this model directly with a Hugging Face pipeline:
22
+ ```python
23
+ import ...
24
+
25
+ base_model_name = "bardsai/jaskier-7b-dpo-v5.6"
26
+ p = pipeline("conversational", model=base_model_name, torch_dtype=torch.float16, device_map="auto")
27
+ messages = [{'messages':[{
28
+ "role": "user",
29
+ "content": "Can Poland into space?"
30
+ }]},
31
+ {'messages':[{
32
+ "role": "user",
33
+ "content": "Tell me where the nearest water park is?"
34
+ }]},
35
+ ]
36
+
37
+ def conversations():
38
+ for x in tqdm(messages):
39
+ yield Conversation(x['messages'])
40
+
41
+ generate_kwargs = {
42
+ "do_sample": True,
43
+ "temperature": 0.7,
44
+ "max_new_tokens": 1024,
45
+ }
46
+ model_outputs = [out[-1]['content'] for out in p(conversations(), **generate_kwargs)]
47
+
48
+ print(model_outputs)
49
+
50
+ ```
51
+
52
+ ## Output
53
+
54
+ >["Poland, as a nation, invests in and collaborates with the international space community, primarily through the Polish Space Agency and the Institute of Space Physics of the Polish Academy of Sciences. They don't physically send Poland to space, but work on advances, technology, and scientific research related to space exploration."]
55
+ >["As an AI, I don't have real-time physical location or personal experiences. I can't provide location-specific information. Please, consider asking a fellow human for such details."]
56
+
57
+ 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.
58
  ## Changelog
59
 
60
  - 2024-02-16: Initial release