gregorkrzmanc commited on
Commit
0407605
·
1 Parent(s): 6005e0f
Files changed (1) hide show
  1. Dockerfile +5 -4
Dockerfile CHANGED
@@ -66,7 +66,7 @@ RUN python3 -m pip install fastjet
66
  RUN python3 -m pip install gradio
67
  RUN python3 -m pip install huggingface_hub
68
  RUN python3 -m pip install hdbscan
69
- RUN python3 -m pip install lgatr
70
  RUN python3 -c "\
71
  from huggingface_hub import snapshot_download; \
72
  snapshot_download(repo_id='gregorkrzmanc/jetclustering', local_dir='models/'); \
@@ -84,9 +84,10 @@ RUN mkdir -p /opt/pepr
84
 
85
 
86
  # Install L-GATr - for some reason this only works if executed from the already-built container
87
- #RUN cd /opt/pepr && git clone https://github.com/gregorkrz/lorentz-gatr lgatr
88
- #RUN cd /opt/pepr/lgatr/ && python3 -m pip install .
89
-
 
90
  # Install torch_cmspepr
91
 
92
  RUN cd /opt/pepr && git clone https://github.com/cms-pepr/pytorch_cmspepr
 
66
  RUN python3 -m pip install gradio
67
  RUN python3 -m pip install huggingface_hub
68
  RUN python3 -m pip install hdbscan
69
+ #RUN python3 -m pip install lgatr # This doesn't work
70
  RUN python3 -c "\
71
  from huggingface_hub import snapshot_download; \
72
  snapshot_download(repo_id='gregorkrzmanc/jetclustering', local_dir='models/'); \
 
84
 
85
 
86
  # Install L-GATr - for some reason this only works if executed from the already-built container
87
+ RUN cd /opt/pepr && git clone https://github.com/gregorkrz/lorentz-gatr lgatr
88
+ RUN cd /opt/pepr/lgatr/ && python3 -m pip install .
89
+ RUN ls /usr/local/lib/python3.10/dist-packages/lgatr
90
+ RUN ls /usr/local/lib/python3.10/dist-packages/lgatr/layers
91
  # Install torch_cmspepr
92
 
93
  RUN cd /opt/pepr && git clone https://github.com/cms-pepr/pytorch_cmspepr