[Feature] add mm token usage (#4570)

* add mm token usage

* fix unit test

* fix unit test

* fix unit test

* fix model path

* fix unit test

* fix unit test

* fix unit test

* remove uncomment

* change var name

* fix code style

* fix code style

* fix code style

* fix code style

* fix unit test
This commit is contained in:
ApplEOFDiscord
2025-10-29 14:37:12 +08:00
committed by GitHub
parent fc5cd1adb1
commit 14f8cddaf1
9 changed files with 70 additions and 20 deletions
+1 -1
View File
@@ -50,7 +50,7 @@ class MockTask:
def get(self, key: str, default_value=None):
if hasattr(self, key):
return getattr(self, key)
elif hasattr(self.sampling_params, key):
elif hasattr(self, "sampling_params") and hasattr(self.sampling_params, key):
return getattr(self.sampling_params, key)
else:
return default_value