Commit
•
4b76d8b
1
Parent(s):
af50340
Remove device args in example code
Browse files
README.md
CHANGED
|
@@ -61,7 +61,6 @@ prompt = processor.tokenizer.apply_chat_template(
|
|
| 61 |
url = "https://www.ilankelman.org/stopsigns/australia.jpg"
|
| 62 |
image = Image.open(requests.get(url, stream=True).raw)
|
| 63 |
inputs = processor(text=prompt, images=image, return_tensors="pt")
|
| 64 |
-
inputs = {k: v.to('cuda') for k, v in inputs.items()}
|
| 65 |
|
| 66 |
# Generate
|
| 67 |
generate_ids = model.generate(**inputs, max_length=30)
|
|
|
|
| 61 |
url = "https://www.ilankelman.org/stopsigns/australia.jpg"
|
| 62 |
image = Image.open(requests.get(url, stream=True).raw)
|
| 63 |
inputs = processor(text=prompt, images=image, return_tensors="pt")
|
|
|
|
| 64 |
|
| 65 |
# Generate
|
| 66 |
generate_ids = model.generate(**inputs, max_length=30)
|