Update .gitlab-ci.yml

.gitlab-ci.yml

@@ -8,7 +8,7 @@
     - docker login -u $CI_REGISTRY_USER -p $CI_REGISTRY_PASSWORD $CI_REGISTRY
     - docker pull $CI_REGISTRY_IMAGE/$TAG || true
     - export DOCKER_BUILDKIT=1
-    - docker build --cache-from $CI_REGISTRY_IMAGE/$TAG -t $CI_REGISTRY_IMAGE/$TAG -f $DOCKERFILE .
+    - docker build --cache-from $CI_REGISTRY_IMAGE/$TAG:$CI_COMMIT_BRANCH -t $CI_REGISTRY_IMAGE/$TAG:$CI_COMMIT_BRANCH -f $DOCKERFILE .
     - docker push $CI_REGISTRY_IMAGE/$TAG:$CI_COMMIT_BRANCH
   # In case we are not on the main branch, we only want to do build testing for
   # the image, without pushing anything to the registry.