entfane
/

gpt2_constitutional_classifier_with_value_head

Text Generation

text-generation-inference

Model card Files Files and versions

entfane commited on Feb 25

Commit

3e877f3

·

verified ·

1 Parent(s): c081146

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -15,8 +15,8 @@ from transformers import AutoTokenizer
 from trl import AutoModelForCausalLMWithValueHead
 import torch
-tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
-model = AutoModelForCausalLMWithValueHead.from_pretrained("openai-community/gpt2", device_map = "cuda")
 messages = [{"role":"system", "content": ""},
               {"role":"user", "content": "How are you doing?"},

 from trl import AutoModelForCausalLMWithValueHead
 import torch
+tokenizer = AutoTokenizer.from_pretrained("entfane/gpt2_constitutional_classifier_with_value_head")
+model = AutoModelForCausalLMWithValueHead.from_pretrained("entfane/gpt2_constitutional_classifier_with_value_head", device_map = "cuda")
 messages = [{"role":"system", "content": ""},
               {"role":"user", "content": "How are you doing?"},