Spaces:

lkjjj26
/

query

Sleeping

lkjjj26 commited on Jan 22

Commit

6fad26e

1 Parent(s): cb73bae

update time app.py cuda

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,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 = """