Replace _target_device with device in semantic search example (#2424)

UKPLab · Jan 18, 2024 · e757296 · e757296
1 parent ed916fb
commit e757296
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/applications/semantic-search/semantic_search_quora_pytorch.py b/examples/applications/semantic-search/semantic_search_quora_pytorch.py
@@ -69,7 +69,7 @@
 print("Corpus loaded with {} sentences / embeddings".format(len(corpus_sentences)))
 
 # Move embeddings to the target device of the model
-corpus_embeddings = corpus_embeddings.to(model._target_device)
+corpus_embeddings = corpus_embeddings.to(model.device)
 
 while True:
     inp_question = input("Please enter a question: ")