Device set to dtype
#2
by
cdreetz
- opened
in ultravox_model.py there is projector.to(config.torch_dtype)
i dont think this is right?
Module.to is overloaded and can take as input a device or a dtype (or in fact a memory_format or a tensor to copy info from): https://pytorch.org/docs/stable/generated/torch.nn.Module.html#torch.nn.Module.to
Does that answer your question or did I misunderstand?