shibbir24 commited on
Commit
2060be0
·
verified ·
1 Parent(s): fce0770

Update evaluate_model.py

Browse files
Files changed (1) hide show
  1. evaluate_model.py +2 -1
evaluate_model.py CHANGED
@@ -12,7 +12,8 @@ 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
+ adapter_path = os.path.join(os.getcwd(), "lora_adapter")
17
  tokenizer = AutoTokenizer.from_pretrained(base_model)
18
  model = AutoModelForCausalLM.from_pretrained(base_model)
19
  model = PeftModel.from_pretrained(model, adapter_path)