Fix code typo for Simple and Video Inference examples (#9)

- Fix code typo for Simple and Video Inference examples (da907c3ab206b8d615c2a4b1b33ae7fd440b4c7d)


Co-authored-by: Michal Kubista <[email protected]>

README.md

@@ -106,7 +106,7 @@ inputs = processor.apply_chat_template(
     tokenize=True,
     return_dict=True,
     return_tensors="pt",
-).to(model.device)
+).to(model.device, dtype=torch.bfloat16)
 
 generated_ids = model.generate(**inputs, do_sample=False, max_new_tokens=64)
 generated_texts = processor.batch_decode(
@@ -137,7 +137,7 @@ inputs = processor.apply_chat_template(
     tokenize=True,
     return_dict=True,
     return_tensors="pt",
-).to(model.device)
+).to(model.device, dtype=torch.bfloat16)
 
 generated_ids = model.generate(**inputs, do_sample=False, max_new_tokens=64)
 generated_texts = processor.batch_decode(