fix(loadModel): pass revision arg too

gadicc · gadicc · commit cd5f995dad91 · 2023-02-21T17:33:32.000Z
diff --git a/api/app.py b/api/app.py
@@ -206,6 +206,7 @@ def inference(all_inputs: dict) -> dict:
                 model_id=normalized_model_id,
                 load=True,
                 precision=model_precision,
+                revision=model_revision,
                 send_opts=send_opts,
             )
             last_model_id = normalized_model_id

Original file line number	Diff line number	Diff line change
`@@ -206,6 +206,7 @@ def inference(all_inputs: dict) -> dict:`
`206`	`206`	`model_id=normalized_model_id,`
`207`	`207`	`load=True,`
`208`	`208`	`precision=model_precision,`
	`209`	`+ revision=model_revision,`
`209`	`210`	`send_opts=send_opts,`
`210`	`211`	`)`
`211`	`212`	`last_model_id = normalized_model_id`