Merge pull request #2 from jwjeffr/main

modified model dir
ovito-org · Jun 6, 2024 · 7171415 · 7171415
2 parents 4f337b4 + 13cb6d3
commit 7171415
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 (