Commit 8d7950f6 authored by zhuww's avatar zhuww
Browse files

modify setup.py

parent 588554dc
include fastfold/common/stereo_chemical_props.txt
...@@ -141,7 +141,7 @@ setup( ...@@ -141,7 +141,7 @@ setup(
description= description=
'Optimizing Protein Structure Prediction Model Training and Inference on GPU Clusters', 'Optimizing Protein Structure Prediction Model Training and Inference on GPU Clusters',
ext_modules=ext_modules, ext_modules=ext_modules,
package_data={'fastfold': ['model/fastnn/kernel/cuda_native/csrc/*']}, package_data={'fastfold': ['model/fastnn/kernel/cuda_native/csrc/*', 'common/stereo_chemical_props.txt']},
cmdclass={'build_ext': BuildExtension} if ext_modules else {}, cmdclass={'build_ext': BuildExtension} if ext_modules else {},
install_requires=['einops', 'colossalai'], install_requires=['einops', 'colossalai'],
) )
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment