Update README.md

README.md

@@ -31,9 +31,6 @@ def get_score(model, tokenizer, prompt, response):
     # Tokenize the input sequences
     inputs = tokenizer.encode_plus(prompt, response, truncation=True, padding="max_length", max_length=512, return_tensors="pt")
 
-    # Move inputs to the appropriate device
-    #inputs = {k: v.to(device) for k, v in inputs.items()}
-
     # Perform forward pass
     with torch.no_grad():
         outputs = model(**inputs)