mishig HF staff commited on
Commit
28d187b
1 Parent(s): c33e089

Remove top_k & repetiton_penalty as they are not supported

Browse files
src/lib/components/InferencePlayground/generationConfigSettings.ts CHANGED
@@ -28,27 +28,13 @@ export const GENERATION_CONFIG_SETTINGS: Record<string, GenerationKeySettings> =
28
  max: 1,
29
  label: "Top-P",
30
  },
31
- top_k: {
32
- default: 50,
33
- step: 1,
34
- min: 1,
35
- max: 100,
36
- label: "Top-K",
37
- },
38
- repetition_penalty: {
39
- default: 1,
40
- step: 0.01,
41
- min: 1,
42
- max: 2,
43
- label: "Repetition Penalty",
44
- },
45
  };
46
 
47
  export type GenerationConfigKey = keyof typeof GENERATION_CONFIG_SETTINGS;
48
 
49
  export const GENERATION_CONFIG_KEYS: GenerationConfigKey[] = ["temperature", "max_tokens"];
50
 
51
- export const GENERATION_CONFIG_KEYS_ADVANCED: GenerationConfigKey[] = ["top_p", "top_k", "repetition_penalty"];
52
 
53
  export type GenerationConfig = Record<GenerationConfigKey, number>;
54
 
 
28
  max: 1,
29
  label: "Top-P",
30
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
31
  };
32
 
33
  export type GenerationConfigKey = keyof typeof GENERATION_CONFIG_SETTINGS;
34
 
35
  export const GENERATION_CONFIG_KEYS: GenerationConfigKey[] = ["temperature", "max_tokens"];
36
 
37
+ export const GENERATION_CONFIG_KEYS_ADVANCED: GenerationConfigKey[] = ["top_p"];
38
 
39
  export type GenerationConfig = Record<GenerationConfigKey, number>;
40