-
Notifications
You must be signed in to change notification settings - Fork 66
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add pip install git+https://gitlab.com/ase/ase
user advice on FrechetCellFilter
ImportError
and allow ase_filter
to be str
#104
Conversation
chgnet/model/dynamics.py
Outdated
@@ -89,7 +89,7 @@ def __init__( | |||
self.device = f"cuda:{cuda_devices_sorted_by_free_mem()[-1]}" | |||
|
|||
# Move the model to the specified device | |||
self.model = (model or CHGNet.load()).to(self.device) | |||
self.model = (model or CHGNet.load().float()).to(self.device) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@BowenD-UCB Not sure it's generally a good idea to lower model precision? But I can't use the model on MPS without it.
Should we restrict the precision lowering to MPS?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Never mind, not actually an issue with CHGNet. Was caused by running MACE and CHGNet in the same session. MACE calls torch.set_default_dtype(torch.float64)
, affecting all other torch models used concurrently.
@ilyes319 Can we make dtype
fully local to MACE to make it compatible with other models?
2aaf67c
to
05a824a
Compare
… CHGNet structural relaxation
… msg on invalid name
05a824a
to
2fb533d
Compare
"should no longer be used. Run pip install git+https://gitlab.com/ase/ase to " | ||
"install from main branch." | ||
) | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@BowenD-UCB We now print this advice when ase.filters.FrechetCellFilter
can't be imported.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
thanks for this!
pip install git+https://gitlab.com/ase/ase
user advice and allow ase_filter
to be str
pip install git+https://gitlab.com/ase/ase
user advice and allow ase_filter
to be str
pip install git+https://gitlab.com/ase/ase
user advice on FrechetCellFilter
ImportError
and allow ase_filter
to be str
chgnet/chgnet/model/dynamics.py
Line 92 in db40e05