Add support for changing soft prompt from userscripts

2025-06-05 21:59:24 +02:00 · 2022-04-12 15:59:05 -04:00
parent 9a2d346d60
commit a3a52dc9c3
3 changed files with 72 additions and 3 deletions
--- a/fileops.py
+++ b/fileops.py
@ -122,7 +122,10 @@ def checksp(filename: str, model_dimension: int) -> Tuple[Union[zipfile.ZipFile,
            shape, fortran_order, dtype = np.lib.format._read_array_header(f, version)
            assert len(shape) == 2
    except:
-        z.close()
+        try:
+            z.close()
+        except UnboundLocalError:
+            pass
        return 1, None, None, None, None
    if dtype not in ('V2', np.float16, np.float32):
        z.close()