shibbir24 commited on
Commit
fce0770
·
verified ·
1 Parent(s): 07c8aec

Update evaluate_model.py

Browse files
Files changed (1) hide show
  1. evaluate_model.py +1 -1
evaluate_model.py CHANGED
@@ -12,7 +12,7 @@ def generate_review(base_model, product, category, features, rating, tone, revie
12
  Generate a product review using LoRA fine-tuned model and apply repetition control.
13
  Optionally evaluates performance every 10 reviews.
14
  """
15
- adapter_path = "./lora_adapter"
16
  tokenizer = AutoTokenizer.from_pretrained(base_model)
17
  model = AutoModelForCausalLM.from_pretrained(base_model)
18
  model = PeftModel.from_pretrained(model, adapter_path)
 
12
  Generate a product review using LoRA fine-tuned model and apply repetition control.
13
  Optionally evaluates performance every 10 reviews.
14
  """
15
+ adapter_path = "lora_adapter"
16
  tokenizer = AutoTokenizer.from_pretrained(base_model)
17
  model = AutoModelForCausalLM.from_pretrained(base_model)
18
  model = PeftModel.from_pretrained(model, adapter_path)