Skip to content

Commit 8e07457

Browse files
authored
Update README.md
1 parent 2fb5977 commit 8e07457

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

README.md

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -16,25 +16,20 @@ https://astroclip.streamlit.app/
1616
The training and evaluation code requires PyTorch 2.0. Additionally, an up-to-date eventlet is required for wandb. Note that the code has only been tested with the specified versions and also expects a Linux environment. To install the AstroCLIP package and its corresponding dependencies, please follow the code below. The install procedure is unfortunately a bit complex, execute the following cell. Do not restart kernel when prompted, instead run the cell below.
1717

1818
```bash
19-
# Base install of required utilities
20-
apt install python3-dev libcairo2-dev pkg-config
21-
2219
# Setting up proper torch version
2320
pip install --upgrade pip
24-
pip install lightning[extra]==2.3.3 boto3==1.28.17
21+
pip install lightning[extra]==2.3.3 boto3==1.28.17
2522
pip install --upgrade pycairo datasets pyarrow
2623
pip install --extra-index-url https://pypi.nvidia.com cuml-cu11
2724
pip install --extra-index-url https://download.pytorch.org/whl/cu117 torch==2.0.0+cu117
28-
pip install torchvision==0.15.0 torchmetrics==0.10.3 dotenv
29-
pip install numpy==1.26.4 --force-reinstall
25+
pip install torchvision==0.15.0 torchmetrics==0.10.3 dotenv numpy==1.26.4
3026

3127
# Installing DiNOv2
3228
pip install omegaconf fvcore iopath
3329
pip install --no-deps git+https://github.com/facebookresearch/dinov2.git@2302b6bf46953431b969155307b9bed152754069
3430

3531
# Installing AstroCLIP
36-
pip install astropy datasets huggingface_hub jaxtyping wandb networkx pyvis
37-
pip uninstall -y transformers
32+
pip install astropy datasets huggingface_hub jaxtyping wandb
3833
pip install --no-deps git+https://github.com/PolymathicAI/AstroCLIP.git
3934
```
4035
**NOTE** The package provides the three shortcuts: `astroclip_trainer` and `spectrum_trainer`, which link to `astroclip/trainer.py`, and `image_trainer`, which links to `astroclip/astrodino/trainer.py`, as long as it is installed. The shortcuts are defined in the `project.scripts` section of the `pyproject.toml` file.

0 commit comments

Comments
 (0)