Spaces:

Hukuna
/

ProteinDesignDemo

Sleeping

Hukuna commited on Feb 25, 2024

Commit

90a3e9a

verified ·

1 Parent(s): 0360759

Update demo.py

Files changed (1) hide show

demo.py CHANGED Viewed

@@ -80,7 +80,7 @@ from chroma.utility.api import register_key
 from chroma.utility.chroma import letter_to_point_cloud, plane_split_protein
 register_key(api_key)
-device = "cpu"
 with contextlib.redirect_stdout(None):
     chroma = Chroma(device=device)

 from chroma.utility.chroma import letter_to_point_cloud, plane_split_protein
 register_key(api_key)
+device = "cuda"
 with contextlib.redirect_stdout(None):
     chroma = Chroma(device=device)