chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,28 @@
|
||||
from transformers import HfArgumentParser
|
||||
|
||||
from FlagEmbedding.evaluation.custom import (
|
||||
CustomEvalArgs, CustomEvalModelArgs,
|
||||
CustomEvalRunner
|
||||
)
|
||||
|
||||
|
||||
def main():
|
||||
parser = HfArgumentParser((
|
||||
CustomEvalArgs,
|
||||
CustomEvalModelArgs
|
||||
))
|
||||
|
||||
eval_args, model_args = parser.parse_args_into_dataclasses()
|
||||
eval_args: CustomEvalArgs
|
||||
model_args: CustomEvalModelArgs
|
||||
|
||||
runner = CustomEvalRunner(
|
||||
eval_args=eval_args,
|
||||
model_args=model_args
|
||||
)
|
||||
|
||||
runner.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user