Spaces:
Sleeping
Sleeping
Update evaluate_model.py
Browse files- 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 |
-
|
| 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)
|