Update ingest.py

per the owner request, default is changed back to "cuda". I run the code on M2 even with default='cuda'. it run somoothly. However, it was typing run on =cuda. if this causes issue, I recommend you to change it to default="mps".
lujingyuanqerq · Jun 3, 2023 · 6668539 · 6668539
1 parent d118003
commit 6668539
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ingest.py b/ingest.py
@@ -38,7 +38,7 @@ def load_documents(source_dir: str) -> List[Document]:
 
 
 @click.command()
-@click.option('--device_type', default='mps', help='device to run on, select gpu, cpu or mps')
+@click.option('--device_type', default='cuda', help='device to run on, select gpu, cpu or mps')
 def main(device_type, ):
     # load the instructorEmbeddings
     if device_type in ['cpu', 'CPU']: