Update README.md

README.md

@@ -15,7 +15,7 @@ Lora weights after finetuning Switch-base-64 on WMT16 datasets.
 from peft import PeftModel
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-base_model = AutoModelForSeq2SeqLM.from_pretrained('google/switch-base-64')
+base_model = AutoModelForSeq2SeqLM.from_pretrained('google/switch-base-64', use_safetensors=False)
 lora_model = PeftModel.from_pretrained(base_model, 'marsggbo/wmt-switch-base-64-lora-ckpt30000')
 merged_model = lora_model.merge_and_unload()
 merged_model.save_pretrained('./switch-base-64-wmt', state_dict=merged_model.state_dict(), safe_serialization=False)