Fix TextEmbeddingModel - Embedding

FBerding · Jul 10, 2023 · 49ddf0c · 49ddf0c
1 parent f5f3fe4
commit 49ddf0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/text_embedding_model.R b/R/text_embedding_model.R
@@ -706,7 +706,7 @@ TextEmbeddingModel<-R6::R6Class(
             for(j in 1:tokens$chunks[i]){
               for(layer in tmp_selected_layer){
                 text_embedding[i,j,]<-text_embedding[i,j,]+as.vector(
-                  reticulate$py_to_r(tensor_embeddings[[as.integer(layer)]][[as.integer(index)]][[as.integer(0)]]$numpy())
+                  tensor_embeddings[[as.integer(layer)]][[as.integer(index)]][[as.integer(0)]]$numpy()
                   )
               }
               text_embedding[i,j,]<-text_embedding[i,j,]/length(tmp_selected_layer)