Pivot / configuration_pivot.py

Upload folder using huggingface_hub

2d8be88 verified about 1 month ago

374 Bytes

"""Configuration for the Pivot set-scored decision encoder."""
from transformers import PretrainedConfig


class PivotConfig(PretrainedConfig):
    model_type = "pivot"

    def __init__(self, dsbt_config=None, encoder_config=None, **kwargs):
        super().__init__(**kwargs)
        self.dsbt_config = dsbt_config or {}
        self.encoder_config = encoder_config or {}

	"""Configuration for the Pivot set-scored decision encoder."""
	from transformers import PretrainedConfig


	class PivotConfig(PretrainedConfig):
	model_type = "pivot"

	def __init__(self, dsbt_config=None, encoder_config=None, **kwargs):
	super().__init__(**kwargs)
	self.dsbt_config = dsbt_config or {}
	self.encoder_config = encoder_config or {}