Update ultravox_model.py

ultravox_model.py

@@ -412,7 +412,10 @@ class UltravoxModel(transformers.LlamaPreTrainedModel, GenerationMixin):
         cls, config: UltravoxConfig
     ) -> "UltravoxProjector":
         projector = UltravoxProjector(config)
-        projector.to(config.torch_dtype)
+        dtype = config.torch_dtype
+        if isinstance(dtype, str):
+            dtype = getattr(torch, dtype)
+        projector.to(dtype)
         return projector
 
     @classmethod