Fix loading weights whose name is UpperCase

frost-beta · Oct 4, 2024 · 1cbc0d1 · 1cbc0d1
1 parent d319c61
commit 1cbc0d1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/lib/utils.ts b/lib/utils.ts
@@ -292,5 +292,8 @@ export function toCamelCase(e: string) {
 }
 
 export function toSnakeCase(e: string) {
+  // Do not convert if the first character is upper case.
+  if (e.length > 0 && e[0] == e[0].toUpperCase())
+    return e;
   return e.replace(/[A-Z]/g, (ch, i) => i ? '_' + ch.toLowerCase() : ch.toLowerCase());
-}
+}