Rename parameter 'ignoreTopKIfOne' to 'ignoreTopKIfZero'

This commit is contained in:
Kwaroran
2024-12-12 10:43:18 +09:00
parent 6d012c77fe
commit fc571d45ac

View File

@@ -96,7 +96,7 @@ type ParameterMap = {
}; };
function applyParameters(data: { [key: string]: any }, parameters: Parameter[], rename: ParameterMap, ModelMode:ModelModeExtended, arg:{ function applyParameters(data: { [key: string]: any }, parameters: Parameter[], rename: ParameterMap, ModelMode:ModelModeExtended, arg:{
ignoreTopKIfOne?:boolean ignoreTopKIfZero?:boolean
} = {}): { [key: string]: any } { } = {}): { [key: string]: any } {
const db = getDatabase() const db = getDatabase()
if(db.seperateParametersEnabled && ModelMode !== 'model'){ if(db.seperateParametersEnabled && ModelMode !== 'model'){
@@ -105,7 +105,7 @@ function applyParameters(data: { [key: string]: any }, parameters: Parameter[],
} }
for(const parameter of parameters){ for(const parameter of parameters){
if(parameter === 'top_k' && arg.ignoreTopKIfOne && db.seperateParameters[ModelMode][parameter] === 1){ if(parameter === 'top_k' && arg.ignoreTopKIfZero && db.seperateParameters[ModelMode][parameter] === 0){
continue continue
} }
@@ -123,7 +123,7 @@ function applyParameters(data: { [key: string]: any }, parameters: Parameter[],
for(const parameter of parameters){ for(const parameter of parameters){
let value = 0 let value = 0
if(parameter === 'top_k' && arg.ignoreTopKIfOne && db.top_k === 1){ if(parameter === 'top_k' && arg.ignoreTopKIfZero && db.top_k === 0){
value = 0 value = 0
} }
switch(parameter){ switch(parameter){
@@ -1505,7 +1505,7 @@ async function requestGoogleCloudVertex(arg:RequestDataArgumentExtended):Promise
'presence_penalty': "presencePenalty", 'presence_penalty': "presencePenalty",
'frequency_penalty': "frequencyPenalty" 'frequency_penalty': "frequencyPenalty"
}, arg.mode, { }, arg.mode, {
ignoreTopKIfOne: true ignoreTopKIfZero: true
}), }),
safetySettings: uncensoredCatagory, safetySettings: uncensoredCatagory,
systemInstruction: { systemInstruction: {