applied-ai-018
/

peacock-data-public-evaluation

Model card Files Files and versions

peacock-data-public-evaluation / Megatron-DeepSpeed /tools /bert_embedding /__init__.py

applied-ai-018's picture

Add files using upload-large-folder tool

ced76ab verified about 1 year ago

history blame contribute delete

127 Bytes

	# Copyright (c) 2023, NVIDIA CORPORATION. All rights reserved.

	from .embed import BertEmbedder, DiskDataParallelBertEmbedder