changing dockerfile

Author: azrael417

docker/Dockerfile

@@ -76,7 +76,7 @@ RUN pip install --ignore-installed "git+https://github.com/NVIDIA/mlperf-common.
 ENV FORCE_CUDA_EXTENSION 1
 ENV TORCH_CUDA_ARCH_LIST "8.0 8.6 9.0 10.0 12.0+PTX"
 ENV HARMONICS_VERSION 0.8.3
-RUN cd /opt && git clone -b tkurth/distributed-attention https://github.com/NVIDIA/torch-harmonics.git && \
+RUN cd /opt && git clone https://github.com/NVIDIA/torch-harmonics.git && \
     cd torch-harmonics && \
     pip install --no-build-isolation -e .