TaimoorSiddiqui commited on
Commit
d2ca7c7
·
1 Parent(s): b1a8ad0

Fix deprecation: torch_dtype → dtype, add import torch

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -55,11 +55,12 @@ tags:
55
  ### Text-only input
56
 
57
  ```python
 
58
  from transformers import AutoModelForImageTextToText, AutoProcessor
59
 
60
  model = AutoModelForImageTextToText.from_pretrained(
61
  "TaimoorSiddiqui/Hopcoder-Mini-9B",
62
- torch_dtype="bfloat16",
63
  device_map="auto",
64
  trust_remote_code=True,
65
  )
@@ -85,7 +86,7 @@ from PIL import Image
85
 
86
  model = AutoModelForImageTextToText.from_pretrained(
87
  "TaimoorSiddiqui/Hopcoder-Mini-9B",
88
- torch_dtype="bfloat16",
89
  device_map="auto",
90
  trust_remote_code=True,
91
  )
 
55
  ### Text-only input
56
 
57
  ```python
58
+ import torch
59
  from transformers import AutoModelForImageTextToText, AutoProcessor
60
 
61
  model = AutoModelForImageTextToText.from_pretrained(
62
  "TaimoorSiddiqui/Hopcoder-Mini-9B",
63
+ dtype=torch.bfloat16,
64
  device_map="auto",
65
  trust_remote_code=True,
66
  )
 
86
 
87
  model = AutoModelForImageTextToText.from_pretrained(
88
  "TaimoorSiddiqui/Hopcoder-Mini-9B",
89
+ dtype=torch.bfloat16,
90
  device_map="auto",
91
  trust_remote_code=True,
92
  )