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 (111a23f9284af5056a970f40d890a8e433b41e84)


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)