Update README.md

README.md

@@ -28,7 +28,7 @@ With a compressed collection of documents to retrieve from, inference becomes ab
 ```python
 from transformers import AutoModel
 
-pisco = AutoModel.from_pretrained('naver/pisco-solar', device_map='cuda')
+pisco = AutoModel.from_pretrained('naver/pisco-solar').to('cuda')
 
 # Example documents and question:
 documents = [