Skip to content

Commit

Permalink
Fix TextEmbeddingModel - Embedding
Browse files Browse the repository at this point in the history
  • Loading branch information
FBerding committed Jul 10, 2023
1 parent f5f3fe4 commit 49ddf0c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion R/text_embedding_model.R
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 49ddf0c

Please sign in to comment.