knc6's picture
Update README.md
7b5cc8b verified
|
raw
history blame
1.18 kB
---
base_model: unsloth/mistral-7b-bnb-4bit
library_name: peft
license: mit
language:
- en
tags:
- chemistry
- text-generation-inference
pipeline_tag: text-generation
---
# Fine-tune models for atomic structure generation of superconductor candidate materials
AtomGPT: Atomistic Generative Pretrained Transformer for Forward and Inverse Materials Design
<!-- Provide a quick summary of what the model is/does. -->
<!-- Provide a longer summary of what this model is. -->
GitHub: [https://github.com/usnistgov/atomgpt](https://github.com/usnistgov/atomgpt)
[Training colab notebook](https://colab.research.google.com/github/knc6/jarvis-tools-notebooks/blob/master/jarvis-tools-notebooks/atomgpt_example.ipynb)
[Inference colab notebook](https://colab.research.google.com/github/knc6/jarvis-tools-notebooks/blob/master/jarvis-tools-notebooks/atomgpt_example_huggingface.ipynb)
Reference: [Choudhary, K. (2024). AtomGPT: Atomistic Generative Pretrained Transformer for Forward and Inverse Materials Design. The Journal of Physical Chemistry Letters, 15, 6909-6917.](https://pubs.acs.org/doi/full/10.1021/acs.jpclett.4c01126)
-
### Framework versions
- PEFT 0.11.1