shibbir24 commited on
Commit
d42abe7
·
verified ·
1 Parent(s): 1e539ba

Update evaluate_model.py

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