Spaces:
Sleeping
Sleeping
Commit
·
0407605
1
Parent(s):
6005e0f
- 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 |
-
|
88 |
-
|
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
|