technicolor
/

TE_Tinyllama

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

technicolor commited on May 14, 2024

Commit

a733ab2

·

verified ·

1 Parent(s): baaa818

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -114,7 +114,8 @@ class TEmbeddingModel(torch.nn.Module):
         return embeddings
 if __name__ == "__main__":
-    TE_model = TEmbeddingModel("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
     TE_model.eval()
     with torch.no_grad():
         output = TE_model(["Hello", "Nice to meet you"])

         return embeddings
 if __name__ == "__main__":
+    # TE_model = TEmbeddingModel("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
+    TE_model = TEmbeddingModel("technicolor/TE_Tinyllama")
     TE_model.eval()
     with torch.no_grad():
         output = TE_model(["Hello", "Nice to meet you"])