diff --git a/app/constants/APIState/LocalAPI.ts b/app/constants/APIState/LocalAPI.ts index d1a0fee..07d2840 100644 --- a/app/constants/APIState/LocalAPI.ts +++ b/app/constants/APIState/LocalAPI.ts @@ -25,6 +25,8 @@ class LocalAPI extends APIBase { { externalName: 'penalty_repeat', samplerID: SamplerID.REPETITION_PENALTY }, { externalName: 'penalty_present', samplerID: SamplerID.PRESENCE_PENALTY }, { externalName: 'penalty_freq', samplerID: SamplerID.FREQUENCY_PENALTY }, + { externalName: 'xtc_t', samplerID: SamplerID.XTC_THRESHOLD }, + { externalName: 'xtc_p', samplerID: SamplerID.XTC_PROBABILITY }, { externalName: 'seed', samplerID: SamplerID.SEED }, ] buildPayload = () => { diff --git a/app/constants/SamplerData.ts b/app/constants/SamplerData.ts index 8f5e68d..53158f3 100644 --- a/app/constants/SamplerData.ts +++ b/app/constants/SamplerData.ts @@ -50,6 +50,9 @@ export enum SamplerID { DRY_BASE = 'dry_base', DRY_ALLOWED_LENGTH = 'dry_allowed_length', DRY_SEQUENCE_BREAK = 'dry_sequence_break', + + XTC_THRESHOLD = 'xtc_threshold', + XTC_PROBABILITY = 'xtc_probability', //... } @@ -160,6 +163,32 @@ export const Samplers = { precision: 2, }, }, + [SamplerID.XTC_PROBABILITY]: { + internalID: SamplerID.XTC_PROBABILITY, + friendlyName: 'XTC Probability', + inputType: 'slider', + values: { + type: 'float', + min: 0, + max: 1, + default: 0, + step: 0.01, + precision: 2, + }, + }, + [SamplerID.XTC_THRESHOLD]: { + internalID: SamplerID.XTC_THRESHOLD, + friendlyName: 'XTC Threshold', + inputType: 'slider', + values: { + type: 'float', + min: 0, + max: 1, + default: 0, + step: 0.01, + precision: 2, + }, + }, [SamplerID.TOP_P]: { internalID: SamplerID.TOP_P, friendlyName: 'Top P', diff --git a/package-lock.json b/package-lock.json index bfb3d76..7c5860a 100644 --- a/package-lock.json +++ b/package-lock.json @@ -24,7 +24,7 @@ "babel-plugin-inline-import": "^3.0.0", "babel-plugin-react-compiler": "^0.0.0-experimental-696af53-20240625", "bert-tokenizer": "^1.1.8", - "cui-llama.rn": "^1.0.11", + "cui-llama.rn": "^1.1.1", "drizzle-orm": "^0.30.10", "eas-cli": "^9.1.0", "expo": "~51.0.28", @@ -14837,9 +14837,9 @@ "license": "MIT" }, "node_modules/cui-llama.rn": { - "version": "1.0.11", - "resolved": "https://registry.npmjs.org/cui-llama.rn/-/cui-llama.rn-1.0.11.tgz", - "integrity": "sha512-8uaXoIbudN93hjC8oHjXap9Zgwuvow1Dsa4SWaQ00Gj4kM/YtHMEFQYRpIMU3GttyqEcR/NsUTbj1d5CXpZu/A==", + "version": "1.1.1", + "resolved": "https://registry.npmjs.org/cui-llama.rn/-/cui-llama.rn-1.1.1.tgz", + "integrity": "sha512-P3q8ZJRybkMuXKJv8knVU8UVyk+7YQQxveUjms+duiuj0NP0JdkT27xMgw4phGGszQsFxmBGhY3xAyZBmZ2jBw==", "license": "MIT", "engines": { "node": ">= 16.0.0" diff --git a/package.json b/package.json index dd582a2..076e4e0 100644 --- a/package.json +++ b/package.json @@ -32,7 +32,7 @@ "babel-plugin-inline-import": "^3.0.0", "babel-plugin-react-compiler": "^0.0.0-experimental-696af53-20240625", "bert-tokenizer": "^1.1.8", - "cui-llama.rn": "^1.0.11", + "cui-llama.rn": "^1.1.1", "drizzle-orm": "^0.30.10", "eas-cli": "^9.1.0", "expo": "~51.0.28",