While loading the model, is it possible to specify to download the pytorch weights instead of the safe tensor weights?
Eg :
pipe = pipeline(“text-generation”, model=“HuggingFaceH4/starchat-beta”, device_map=“auto”)
When the above line is executed, it downloads the safe tensor weights by default, is it possible to direct it to download the pytorch weights.