Update README.md

README.md

@@ -37,7 +37,7 @@ def response_generate(
         return_token_type_ids=False,
     ).to(
         device
-    )  # tokenize inputs, load on device
+    )
 
     with torch.autocast("cuda", dtype=torch.bfloat16):
         response = model.generate(