diff --git a/apps/shark_studio/api/sd.py b/apps/shark_studio/api/sd.py index 8499b652dc..4fbabdb33c 100644 --- a/apps/shark_studio/api/sd.py +++ b/apps/shark_studio/api/sd.py @@ -103,6 +103,9 @@ def __init__( self.height = height self.width = width self.scheduler_obj = {} + + # TODO: Reinstate commented out *kwargs once supported by the relevant + # turbine.models static_kwargs = { "pipe": { "external_weights": "safetensors", @@ -130,7 +133,7 @@ def __init__( "batch_size": batch_size, "height": height, "width": width, - "precision": precision, + # "precision": precision, }, "vae_decode": { "hf_model_name": base_model_id, @@ -141,7 +144,7 @@ def __init__( "batch_size": batch_size, "height": height, "width": width, - "precision": precision, + # "precision": precision, }, } super().__init__(sd_model_map, base_model_id, static_kwargs, device, import_ir)