add comment to cite tei project

predibase · Apr 12, 2024 · c3ddb44 · c3ddb44
1 parent b24dc3e
commit c3ddb44
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/server/lorax_server/models/flash_bert.py b/server/lorax_server/models/flash_bert.py
@@ -24,6 +24,11 @@
 tracer = trace.get_tracer(__name__)
 
 
+# NOTE: This implementation of flashbert was based on the huggingface/text-embeddings-inference implementation of flashbert here:
+# https://github.com/huggingface/text-embeddings-inference/blob/cb802a25d43fe6078c715b49652a3bc8a7d5aac8/backends/python/server/text_embeddings_server/models/flash_bert.py
+
+
+
 
 class BertEmbeddings:
     def __init__(self, prefix, weights, device, dtype, config: BertConfig):