fix(loadModel): revision = None if revision == "" else revision

gadicc · gadicc · commit 1773631e292e · 2023-01-09T12:13:22.000+02:00
diff --git a/api/loadModel.py b/api/loadModel.py
@@ -26,6 +26,9 @@
 
 def loadModel(model_id: str, load=True, precision=None, revision=None):
     torch_dtype = torch_dtype_from_precision(precision)
+    if revision == "":
+        revision = None
+
     print(
         "loadModel",
         {

Original file line number	Diff line number	Diff line change
`@@ -26,6 +26,9 @@`
`26`	`26`
`27`	`27`	`def loadModel(model_id: str, load=True, precision=None, revision=None):`
`28`	`28`	`torch_dtype = torch_dtype_from_precision(precision)`
	`29`	`+ if revision == "":`
	`30`	`+ revision = None`
	`31`	`+`
`29`	`32`	`print(`
`30`	`33`	`"loadModel",`
`31`	`34`	`{`