theblackcat102 commited on
Commit
a71ef5b
1 Parent(s): 2b7f7d6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -69,8 +69,8 @@ model = AutoModelForCausalLM.from_pretrained(model_name).half().eval().cuda()
69
 
70
  input_text = """
71
  <|startoftoken|>system
72
- You are a helpful assistant<|endoftoken|><startoftoken>human
73
- What's the population of the earth?<|endoftoken|><startoftoken>assistant
74
 
75
  """
76
  inputs = tokenizer(input_text, return_tensors="pt", padding=True).to(0)
 
69
 
70
  input_text = """
71
  <|startoftoken|>system
72
+ You are a helpful assistant<|endoftoken|><|startoftoken|>human
73
+ What's the population of the earth?<|endoftoken|><|startoftoken|>assistant
74
 
75
  """
76
  inputs = tokenizer(input_text, return_tensors="pt", padding=True).to(0)