Delete hf_model.py

hf_model.py

@@ -1,44 +0,0 @@
-from typing import Any, Sequence
-
-from transformers import PreTrainedModel, PretrainedConfig
-import hydra
-import torch
-
-from models.flow_matching import DummyContentAudioFlowMatching
-
-
-class UniFlowAudioConfig(PretrainedConfig):
-    def __init__(
-        self,
-        config: dict = {},
-        **kwargs,
-    ):
-        self.config = config
-        super().__init__(**kwargs)
-
-
-class UniFlowAudioModel(PreTrainedModel):
-    config_class = UniFlowAudioConfig
-
-    def __init__(self, config: UniFlowAudioConfig):
-        super().__init__(config)
-        self.model: DummyContentAudioFlowMatching = hydra.utils.instantiate(
-            config.config, _convert_="all"
-        )
-
-    def forward(
-        self,
-        content: list[Any],
-        task: list[str],
-        is_time_aligned: Sequence[bool],
-        instruction: torch.Tensor,
-        instruction_lengths: Sequence[int],
-        num_steps: int = 20,
-        sway_sampling_coef: float | None = -1.0,
-        guidance_scale: float = 3.0,
-        disable_progress: bool = True,
-    ):
-        return self.model.inference(
-            content, task, is_time_aligned, instruction, instruction_lengths,
-            num_steps, sway_sampling_coef, guidance_scale, disable_progress
-        )