Files
shap-e/shap_e/diffusion/sample.py