From 6caac014940bfe1bbe710bec6cd21a97c3b4956d Mon Sep 17 00:00:00 2001 From: Roy Han Date: Wed, 3 Jul 2024 14:05:34 -0700 Subject: [PATCH] clear comments --- llm/server.go | 1 - server/sched_test.go | 4 ---- 2 files changed, 5 deletions(-) diff --git a/llm/server.go b/llm/server.go index db4431f56..80eaab089 100644 --- a/llm/server.go +++ b/llm/server.go @@ -33,7 +33,6 @@ type LlamaServer interface { Ping(ctx context.Context) error WaitUntilRunning(ctx context.Context) error Completion(ctx context.Context, req CompletionRequest, fn func(CompletionResponse)) error - // Embedding(ctx context.Context, prompt string) ([]float64, error) Embed(ctx context.Context, input []string) ([][]float32, error) Tokenize(ctx context.Context, content string) ([]int, error) Detokenize(ctx context.Context, tokens []int) (string, error) diff --git a/server/sched_test.go b/server/sched_test.go index e3b1506ba..c8930fcf8 100644 --- a/server/sched_test.go +++ b/server/sched_test.go @@ -648,10 +648,6 @@ func (s *mockLlm) WaitUntilRunning(ctx context.Context) error { return s.waitRes func (s *mockLlm) Completion(ctx context.Context, req llm.CompletionRequest, fn func(llm.CompletionResponse)) error { return s.completionResp } - -// func (s *mockLlm) Embedding(ctx context.Context, prompt string) ([]float64, error) { -// return s.embeddingResp, s.embeddingRespErr -// } func (s *mockLlm) Embed(ctx context.Context, input []string) ([][]float32, error) { return s.embedResp, s.embedRespErr }