Update README.md
Browse files
README.md
CHANGED
|
@@ -77,7 +77,7 @@ To use the model with the Hugging Face `transformers` library:
|
|
| 77 |
from transformers import AutoProcessor, Idefics3ForConditionalGeneration
|
| 78 |
|
| 79 |
DEVICE = f"cuda" if torch.cuda.is_available() else 'cpu' if torch.cpu.is_available() else 'mps'
|
| 80 |
-
|
| 81 |
if DEVICE == 'cuda': display(torch.cuda.device_count())
|
| 82 |
|
| 83 |
N = 5
|
|
|
|
| 77 |
from transformers import AutoProcessor, Idefics3ForConditionalGeneration
|
| 78 |
|
| 79 |
DEVICE = f"cuda" if torch.cuda.is_available() else 'cpu' if torch.cpu.is_available() else 'mps'
|
| 80 |
+
print(DEVICE)
|
| 81 |
if DEVICE == 'cuda': display(torch.cuda.device_count())
|
| 82 |
|
| 83 |
N = 5
|