Downloads · 30 days
0
Akshayuc16/QuPep-GenAI
QuPep-GenAI is a machine learning model from Akshayuc16. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This repository is intended to host a trained checkpoint (model.pt) for the energy-aware VAE used to sample IDR conformational ensembles.
Downloads · 30 days
0
Access
Public
Updated Jan 9, 2026
Repo size
5.2 MB
Likes
1
Public
Click a slice to open those files.
.pt5.2 MB · 100%
From the Hugging Face model README
This repository is intended to host a trained checkpoint (model.pt) for the energy-aware VAE used to sample IDR conformational ensembles.
model.pt: PyTorch checkpoint produced by the training step in idr_quantum_genai_pipeline.py.inference.py: Inference-only loader + sampler utilities (no Qiskit dependencies).python -c "import torch; from inference import load_model_from_checkpoint, sample_ensemble_arrays; m=load_model_from_checkpoint('model.pt','cpu'); out=sample_ensemble_arrays(m,'DSKRETYS',n_samples=100); print(out['energy'][:5], out['angles'].shape)"
n_samples to avoid timeouts.