Spaces:

lkjjj26
/

query

Sleeping

lkjjj26 commited on Jan 22

Commit

adaf6c5

1 Parent(s): 422b4c9

update app.py cuda

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ class PDBSearchAssistant:
             max_new_tokens=1024,
             temperature=0.1,
             torch_dtype="auto",
-            device="cpu" # cuda or cpu
         )
         self.prompt_template = """

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