feat: added xtc sampling to local

Vali-98 · Aug 21, 2024 · 746f401 · 746f401
1 parent 52cf9c9
commit 746f401
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 5 deletions.
diff --git 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
@@ -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
diff --git 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",