Browse Source

adding back Ed's REVISION when assigning the self.fine_tuned_model

pull/319/head
craigprobus 4 weeks ago
parent
commit
c9564a647a
  1. 2
      week8/pricer_service2.py

2
week8/pricer_service2.py

@ -58,7 +58,7 @@ class Pricer:
quantization_config=quant_config, quantization_config=quant_config,
device_map="auto" 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() @modal.method()
def price(self, description: str) -> float: def price(self, description: str) -> float:

Loading…
Cancel
Save