Update README.md

README.md

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