Update instructor.py

Thanks for @racinmat for reminding me to change this.
xlang-ai · Aug 24, 2024 · 9b8bb58 · 9b8bb58
1 parent f7d6914
commit 9b8bb58
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/InstructorEmbedding/instructor.py b/InstructorEmbedding/instructor.py
@@ -623,7 +623,7 @@ def encode(
             input_was_string = True
 
         if device is None:
-            device = self._target_device
+            device = self.device
 
         self.to(device)
 
@@ -690,4 +690,4 @@ def encode(
         if input_was_string:
             all_embeddings = all_embeddings[0]
 
-        return all_embeddings
+        return all_embeddings