download.sh

cd /home/user/app
mkdir aglib
cd aglib
git clone https://github.com/tpgh24/ag4masses.git

pip cache purge
pip install --upgrade pip
pip install --upgrade packaging setuptools setuptools_scm wheel 
# pip install typing_extensions==4.6.0
pip install --require-hashes --no-deps -r /home/user/app/aglib/ag4masses/alphageometry/requirements.txt
pip install typing_extensions==4.6.0

# cd alphageometry
git clone https://github.com/google-research/meliad.git

# cd ..
# mkdir aglib
git lfs install
git clone https://huggingface.co/Wauplin/alphageometry ag_ckpt_vocab
export DATA=ag_ckpt_vocab

cd /home/user/app
# some patch for cpu
# cp models.py /home/user/app/aglib/ag4masses/alphageometry/models.py
# cp lm_inference.py /home/user/app/aglib/ag4masses/alphageometry/lm_inference.py
# cp base_htrans.gin /home/user/app/aglib/meliad/transformer/configs/base_htrans.gin
cp alphageometry.py /home/user/app/aglib/ag4masses/alphageometry/alphageometry.py
cp numericals.py /home/user/app/aglib/ag4masses/alphageometry/numericals.py
cp ag4masses-public.ipynb /data/ag4masses-public.ipynb
cp problems.txt /data/problems.txt
mkdir /data/ag4mtest