Fix embeddings.

Ai00-X · Jul 22, 2023 · b836bfe · b836bfe
1 parent 5a16143
commit b836bfe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/embedding.rs b/src/embedding.rs
@@ -16,7 +16,7 @@ impl From<EmbeddingRequest> for GenerateRequest {
     fn from(value: EmbeddingRequest) -> Self {
         Self {
             prompt: Vec::from(value.input).join(""),
-            max_tokens: 0,
+            max_tokens: 1,
             stop: Default::default(),
             sampler: Default::default(),
             occurrences: Default::default(),