DRXD1000
/

Phoenix-7B

Text Generation

alignment-handbook

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

DRXD1000 commited on Feb 12, 2024

Commit

4d2310a

•

1 Parent(s): 313d20b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -109,7 +109,7 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 model = AutoModelForCausalLM.from_pretrained("DRXD1000/Phoenix", torch_dtype=torch.bfloat16, device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained("DRXD1000/Phoenix")
-prompt = messages = [
     {
         "role": "system",
         "content": "", #Not recommended. Phoenix does not react well on system prompts

 from transformers import AutoModelForCausalLM, AutoTokenizer
 model = AutoModelForCausalLM.from_pretrained("DRXD1000/Phoenix", torch_dtype=torch.bfloat16, device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained("DRXD1000/Phoenix")
+prompt = [
     {
         "role": "system",
         "content": "", #Not recommended. Phoenix does not react well on system prompts