Sharathhebbar24 commited on
Commit
2c35dac
·
verified ·
1 Parent(s): 9687554

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +51 -0
README.md ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ datasets:
4
+ - HuggingFaceH4/ultrachat_200k
5
+ - Intel/orca_dpo_pairs
6
+ language:
7
+ - en
8
+ pipeline_tag: text-generation
9
+ tags:
10
+ - gpt2
11
+ - dpo
12
+ ---
13
+ This model is a finetuned version of ```Sharathhebbar24/chat_gpt2``` using ```Intel/orca_dpo_pairs``` on DPO
14
+
15
+ ## Model description
16
+
17
+ GPT-2 is a transformers model pre-trained on a very large corpus of English data in a self-supervised fashion. This
18
+ means it was pre-trained on the raw texts only, with no humans labeling them in any way (which is why it can use lots
19
+ of publicly available data) with an automatic process to generate inputs and labels from those texts. More precisely,
20
+ it was trained to guess the next word in sentences.
21
+
22
+ More precisely, inputs are sequences of continuous text of a certain length and the targets are the same sequence,
23
+ shifting one token (word or piece of word) to the right. The model uses a masking mechanism to make sure the
24
+ predictions for the token `i` only use the inputs from `1` to `i` but not the future tokens.
25
+
26
+ This way, the model learns an inner representation of the English language that can then be used to extract features
27
+ useful for downstream tasks. The model is best at what it was trained for, however, which is generating texts from a
28
+ prompt.
29
+
30
+ ### To use this model
31
+
32
+ ```python
33
+ >>> from transformers import AutoTokenizer, AutoModelForCausalLM
34
+ >>> model_name = "Sharathhebbar24/chat_gpt2_dpo"
35
+ >>> model = AutoModelForCausalLM.from_pretrained(model_name)
36
+ >>> tokenizer = AutoTokenizer.from_pretrained(model_name)
37
+ >>> def generate_text(prompt):
38
+ >>> inputs = tokenizer.encode(prompt, return_tensors='pt')
39
+ >>> outputs = model.generate(inputs, max_length=64, pad_token_id=tokenizer.eos_token_id)
40
+ >>> generated = tokenizer.decode(outputs[0], skip_special_tokens=True)
41
+ >>> return generated[:generated.rfind(".")+1]
42
+ >>> prompt = """
43
+ >>> user: what are you?
44
+ >>> assistant: I am a Chatbot intended to give a python program
45
+ >>> user: hmm, can you write a python program to print Hii Heloo
46
+ >>> assistant: Sure Here is a python code.\n print("Hii Heloo")
47
+ >>> user: Can you write a Linear search program in python
48
+ >>> """
49
+ >>> res = generate_text(prompt)
50
+ >>> res
51
+ ```