maybe fix 2
Browse files
models/matany_compat_patch.py
CHANGED
|
@@ -59,7 +59,7 @@ def method_compat(self, img, *args, **kwargs):
|
|
| 59 |
return orig_method(self, img, *args, **kwargs)
|
| 60 |
|
| 61 |
setattr(MatAnyone, method_name, method_compat)
|
| 62 |
-
setattr(
|
| 63 |
log.info(f"[MatAnyCompat] Applied T=1 squeeze guard in MatAnyone.{method_name}")
|
| 64 |
patched = True
|
| 65 |
|
|
|
|
| 59 |
return orig_method(self, img, *args, **kwargs)
|
| 60 |
|
| 61 |
setattr(MatAnyone, method_name, method_compat)
|
| 62 |
+
setattr(MatAnyone, f"_{method_name}_patched", True)
|
| 63 |
log.info(f"[MatAnyCompat] Applied T=1 squeeze guard in MatAnyone.{method_name}")
|
| 64 |
patched = True
|
| 65 |
|