Remove module, fix code

Author: matgille

aquilign/align/bertalign/Bertalign.py

@@ -390,16 +390,12 @@ def find_top_k_sents(src_vecs, tgt_vecs, k=3, device='cpu'):
     """
     embedding_size = src_vecs.shape[1]
     if torch.cuda.is_available() and platform == 'linux' and device != "cpu": # GPU version
-        print("OK")
-        exit(0)
         res = faiss.StandardGpuResources() 
         index = faiss.IndexFlatIP(embedding_size)
         gpu_index = faiss.index_cpu_to_gpu(res, 0, index)
         gpu_index.add(tgt_vecs) 
         D, I = gpu_index.search(src_vecs, k)
     else: # CPU version
-        print("Not OK")
-        exit(0)
         index = faiss.IndexFlatIP(embedding_size)
         index.add(tgt_vecs)
         D, I = index.search(src_vecs, k)