Fix: move inputs to model device in inference example to avoid "same device" error (#3)

- Fix: move inputs to model device in inference example to avoid "same device" error (c73895a07ad38c9a21be7bc68bfcdd556370eb17)


Co-authored-by: Lim Seong Geun <lim4349@users.noreply.huggingface.co>

README.md

@@ -116,6 +116,7 @@ inputs = processor.apply_chat_template(
     return_dict=True,
     return_tensors="pt"
 )
+inputs = inputs.to(model.device)
 
 # Inference: Generation of the output
 generated_ids = model.generate(**inputs, max_new_tokens=128)