From c9564a647adfb67a835444bfe5851bdcfac64feb Mon Sep 17 00:00:00 2001 From: craigprobus Date: Sun, 13 Apr 2025 15:19:58 -0700 Subject: [PATCH] adding back Ed's REVISION when assigning the self.fine_tuned_model --- week8/pricer_service2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/week8/pricer_service2.py b/week8/pricer_service2.py index 51731d7..1f46dc4 100644 --- a/week8/pricer_service2.py +++ b/week8/pricer_service2.py @@ -58,7 +58,7 @@ class Pricer: quantization_config=quant_config, device_map="auto" ) - self.fine_tuned_model = PeftModel.from_pretrained(self.base_model, FINETUNED_MODEL) + self.fine_tuned_model = PeftModel.from_pretrained(self.base_model, FINETUNED_MODEL, revision=REVISION) @modal.method() def price(self, description: str) -> float: