Spaces:

lkjjj26
/

query

Sleeping

lkjjj26 commited on Jan 10

Commit

1742853

verified ·

1 Parent(s): 039581d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ class PDBSearchAssistant:
             max_new_tokens=1024,
             temperature=0.1,
             torch_dtype="auto",
-            device="mps"
         )
         self.prompt_template = """

             max_new_tokens=1024,
             temperature=0.1,
             torch_dtype="auto",
+            device="cuda"
         )
         self.prompt_template = """