Praveen0309 commited on
Commit
9732f39
1 Parent(s): 1c98b64
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -34,8 +34,8 @@ peft_lora_adapter = PeftModel.from_pretrained(base_model, peft_lora_adapter_path
34
  base_model.load_adapter(peft_lora_adapter_path, adapter_name="lora_adapter")
35
 
36
  processor = AutoProcessor.from_pretrained("HuggingFaceH4/vsft-llava-1.5-7b-hf-trl")
37
- model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
38
- tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M")
39
 
40
 
41
  # model_id = r"C:\Users\prave\OneDrive\Desktop\MLOPS\Mlops_2\huggingface_model"
 
34
  base_model.load_adapter(peft_lora_adapter_path, adapter_name="lora_adapter")
35
 
36
  processor = AutoProcessor.from_pretrained("HuggingFaceH4/vsft-llava-1.5-7b-hf-trl")
37
+ # model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_418M")
38
+ # tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_418M")
39
 
40
 
41
  # model_id = r"C:\Users\prave\OneDrive\Desktop\MLOPS\Mlops_2\huggingface_model"