Update ultravox_config.py

ultravox_config.py

@@ -1,5 +1,4 @@
 import dataclasses
-import torch
 from enum import Enum
 from typing import Any, Dict, List, Optional
 
@@ -116,12 +115,8 @@ class UltravoxConfig(transformers.PretrainedConfig):
         text_model_lora_config: LoraConfigSimplified | None = None,
         audio_model_lora_config: LoraConfigSimplified | None = None,
         audio_latency_block_size: int | None = None,
-        torch_type: str | None = None,
         **kwargs,
     ):
-        if isinstance(torch_type, str):
-            torch_type = getattr(torch, torch_type)
-
         self.ignore_index = ignore_index
 
         self.audio_model_id = audio_model_id
@@ -169,7 +164,7 @@ class UltravoxConfig(transformers.PretrainedConfig):
 
         self.initializer_range = text_config.initializer_range
 
-        super().__init__(**kwargs, torch_type=torch_type)
+        super().__init__(**kwargs)
 
     def to_diff_dict(self) -> Dict[str, Any]:
         diff_dict = super().to_diff_dict()