diff --git a/egs/librispeech/ASR/.distillation_with_hubert.sh.swp b/egs/librispeech/ASR/.distillation_with_hubert.sh.swp
index 3d1134647..e22b91bb5 100644
Binary files a/egs/librispeech/ASR/.distillation_with_hubert.sh.swp and b/egs/librispeech/ASR/.distillation_with_hubert.sh.swp differ
diff --git a/egs/librispeech/ASR/distillation_with_hubert.sh b/egs/librispeech/ASR/distillation_with_hubert.sh
index dc153afbc..9a7c3a1df 100755
--- a/egs/librispeech/ASR/distillation_with_hubert.sh
+++ b/egs/librispeech/ASR/distillation_with_hubert.sh
@@ -171,7 +171,8 @@ if [ $stage -le 2 ] && [ $stop_stage -ge 2 ]; then
     log "Remove $codebook_download_dir"
     rm -rf $codebook_download_dir
   fi
-
+  
+  CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
   ./pruned_transducer_stateless6/extract_codebook_index.py \
     --full-libri $full_libri \
     --exp-dir $exp_dir \