Update merge_lora_into_ggml.py

RWKV · Feb 25, 2024 · a668650 · a668650
1 parent cb74c44
commit a668650
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/merge_lora_into_ggml.py b/python/merge_lora_into_ggml.py
@@ -61,9 +61,9 @@ def main() -> None:
         header: Tuple[int, int, int, int, int, int] = struct.unpack('=iiiiii', in_file.read(6 * 4))
 
         if header[0] != 0x67676d66:
-            raise ValueError(f'Invalid magic value: {header[0]:x}')
-        if not 100 <= header[1] <= 101:
-            raise ValueError(f'Invalid version number: {header[1]}')
+            raise ValueError(f'Invalid magic value {header[0]:x}')
+        if not (100 <= header[1] <= 101):
+            raise ValueError(f'Invalid version number {header[1]}')
         if not (header[5] == 0 or header[5] == 1):
             raise ValueError('Only FP32 and FP16 models are supported')