소스 검색

clear comments

Roy Han 10 달 전
부모
커밋
6caac01494
2개의 변경된 파일0개의 추가작업 그리고 5개의 파일을 삭제
  1. 0 1
      llm/server.go
  2. 0 4
      server/sched_test.go

+ 0 - 1
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)

+ 0 - 4
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
 }