Update README.md

README.md

@@ -45,7 +45,7 @@ model = AutoModel.from_pretrained("yale-nlp/MDCureRM").to(torch.device("cuda"))
 tokenizer = AutoTokenizer.from_pretrained("yale-nlp/MDCureRM", use_fast=True)
 tokenizer.pad_token = tokenizer.eos_token
 
-reward_weights = torch.tensor([1/9, 1/9, 2/9, 2/9, 2/9, 2/9], device="cuda")
+reward_weights = torch.tensor([1/9, 1/9, 1/9, 2/9, 2/9, 2/9], device="cuda")
 
 source_text_1 = ...
 source_text_2 = ...