mishig HF staff commited on
Commit
15d6d38
1 Parent(s): c73883a

Imrpove types

Browse files
Files changed (1) hide show
  1. src/lib/types/index.d.ts +3 -3
src/lib/types/index.d.ts CHANGED
@@ -3,7 +3,7 @@ import type { ChatCompletionInputMessage } from '@huggingface/tasks';
3
 
4
  type Model = string;
5
 
6
- type ModelConfig = {
7
  temperature: number;
8
  maxTokens: number;
9
  streaming: boolean;
@@ -12,8 +12,8 @@ type ModelConfig = {
12
 
13
  type Conversation = {
14
  id: string;
15
- model: ModelEntry;
16
- config: ModelConfig;
17
  messages: ChatCompletionInputMessage[];
18
  };
19
 
 
3
 
4
  type Model = string;
5
 
6
+ type GenerationConfig = {
7
  temperature: number;
8
  maxTokens: number;
9
  streaming: boolean;
 
12
 
13
  type Conversation = {
14
  id: string;
15
+ model: ModelEntryWithTokenizer;
16
+ config: GenerationConfig;
17
  messages: ChatCompletionInputMessage[];
18
  };
19