Update test_generator.py

globus-labs · Nov 8, 2023 · 06f7842 · 06f7842
1 parent 212bda1
commit 06f7842
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_generator.py b/tests/test_generator.py
@@ -8,7 +8,7 @@
 
 @fixture()
 def load_model():
-    model = 
+    model = "mofa/models/geom_difflinker.ckpt"
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     ddpm = DDPM.load_from_checkpoint(model, map_location=device).eval().to(device)
     return ddpm