modified model dir

ovito-org · Apr 30, 2024 · 13cb6d3 · 13cb6d3
1 parent 4f337b4
commit 13cb6d3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/scoreBasedDenoising/__init__.py b/src/scoreBasedDenoising/__init__.py
@@ -109,7 +109,8 @@ def getModelPath(self):
             if not path.exists():
                 raise FileNotFoundError(f"{path} does not exist.")
             return Path(self.model_path)
-        modelDir = impRes.files("graphite.pretrained_models.denoiser")
+        # modelDir = impRes.files("graphite.pretrained_models.denoiser")
+        modelDir = impRes.files("graphite") / "pretrained_models" / "denoiser"
         if self.structure == "SiO2":
             return modelDir.joinpath("SiO2-denoiser.pt")
         elif (