-
Notifications
You must be signed in to change notification settings - Fork 11
Expand file tree
/
Copy path__init__.py
More file actions
28 lines (23 loc) · 811 Bytes
/
__init__.py
File metadata and controls
28 lines (23 loc) · 811 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
try:
from .gemini_verifier import GeminiVerifier
except Exception as e:
GeminiVerifier = None
try:
from .openai_verifier import OpenAIVerifier
except Exception as e:
OpenAIVerifier = None
try:
from .qwen_verifier import QwenVerifier
except Exception as e:
QwenVerifier = None
try:
from .laion_aesthetics import LAIONAestheticVerifier
except Exception as e:
LAIONAestheticVerifier = None
SUPPORTED_VERIFIERS = {
"qwen": QwenVerifier if QwenVerifier else None,
"gemini": GeminiVerifier if GeminiVerifier else None,
"openai": OpenAIVerifier if OpenAIVerifier else None,
"laion_aesthetic": LAIONAestheticVerifier if LAIONAestheticVerifier else None,
}
SUPPORTED_METRICS = {k: getattr(v, "SUPPORTED_METRIC_CHOICES", None) for k, v in SUPPORTED_VERIFIERS.items()}