We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a5731e9 commit 9df7790Copy full SHA for 9df7790
nff/nn/models/mace.py
@@ -294,10 +294,8 @@ def load_foundations(
294
NffScaleMACE: NffScaleMACE foundational model.
295
"""
296
mace_model_path = get_mace_foundtion_model_path(model)
297
- print(mace_model_path)
298
mace_model = torch.load(mace_model_path, map_location=map_location, weights_only=False)
299
init_params = get_init_kwargs_from_model(mace_model)
300
- print(init_params)
301
model_dtype = get_model_dtype(mace_model)
302
if default_dtype == "":
303
if not suppress_warnings:
0 commit comments