legalkit-retrieval

Sleeping

louisbrulenaudet commited on Apr 7, 2024

Commit

5366b26

verified ·

1 Parent(s): 6b2dcd4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -112,7 +112,7 @@ Additionally, the binary index is much faster (up to 32x) to search than the flo
 instance, dataset, dataframe, DESCRIPTION = setup(
     model_name="louisbrulenaudet/tsdae-lemone-mbert-base",
     description=DESCRIPTION,
-    device="cuda",
     ndim=768,
     metric="ip",
     dtype="i8"

 instance, dataset, dataframe, DESCRIPTION = setup(
     model_name="louisbrulenaudet/tsdae-lemone-mbert-base",
     description=DESCRIPTION,
+    device="cpu",
     ndim=768,
     metric="ip",
     dtype="i8"