| import os |
|
|
|
|
| MATANYONE_V1 = "v1" |
| MATANYONE_V2 = "v2" |
| MATANYONE2_SIGNATURE_OFFSET = 61576 |
| MATANYONE1_SIGNATURE_BYTES = bytes([23, 65, 57, 61, 52, 154, 201, 60, 231, 125, 180, 188, 187, 85, 166, 61]) |
| MATANYONE2_SIGNATURE_BYTES = bytes([155, 244, 100, 61, 100, 179, 194, 60, 92, 239, 124, 186, 254, 127, 168, 61]) |
|
|
|
|
| def detect_matanyone_model_version(model_path): |
| if model_path is None or not os.path.isfile(model_path): |
| return None |
|
|
| with open(model_path, "rb") as reader: |
| reader.seek(MATANYONE2_SIGNATURE_OFFSET) |
| actual = reader.read(len(MATANYONE2_SIGNATURE_BYTES)) |
|
|
| if actual == MATANYONE1_SIGNATURE_BYTES: |
| return MATANYONE_V1 |
| if actual == MATANYONE2_SIGNATURE_BYTES: |
| return MATANYONE_V2 |
| return None |
|
|
|
|
| def delete_if_not_matanyone2_model(model_path): |
| if detect_matanyone_model_version(model_path) is not None: |
| return False |
|
|
| os.remove(model_path) |
| return True |
|
|
