Minor bug fixes

inference.py

@@ -892,7 +892,10 @@ def generate_linkers(
             linker_ckpt = linker_ckpt['pocket_full']
     else:
         linker_ckpt = linker_ckpt['geom']
-    ddpm = DDPM.load_from_checkpoint(linker_ckpt, robust=robust, map_location=device).eval().to(device)
+    ddpm = DDPM.load_from_checkpoint(
+        linker_ckpt,
+        robust=robust, torch_device=device, map_location=device
+    ).eval().to(device)
     is_geom = ddpm.is_geom
 
     if random_seed is not None: