Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -339,7 +339,7 @@ model = AutoModelForVision2Seq.from_pretrained(
339
  "HuggingFaceM4/idefics2-8b",
340
  + torch_dtype=torch.float16,
341
  + quantization_config=quantization_config,
342
- ).to(DEVICE)
343
  ```
344
 
345
  </details>
 
339
  "HuggingFaceM4/idefics2-8b",
340
  + torch_dtype=torch.float16,
341
  + quantization_config=quantization_config,
342
+ )
343
  ```
344
 
345
  </details>