antonypamo/RRF

trained_icosahedral_rrf_model

Contents

• model.py : extracted model class definitions
• inference.py : load and run the model with PyTorch
• demo.ipynb : quick notebook demo for inference
• requirements.txt : dependencies
• configs/config.json : hyperparameters/config template
• examples/ : sample usage scripts
• scripts/upload_to_hf.py : helper to push model to the Hub

Usage

pip install -r requirements.txt
python inference.py

Replace YourModelClass in inference.py with the actual class defined in model.py.