From f9c15eef044695b2fced23b35a1f42a84ed15ec6 Mon Sep 17 00:00:00 2001 From: one-lithe-rune Date: Sun, 25 Feb 2024 09:10:57 +0000 Subject: [PATCH] Studio2/SD/API: Exclude more static_kwargs for now - Comment out precision being set in static_kwargs for vae_encode when setting up a StableDiffusion pipeline - Comment out precision being set in static_kwargs for vae_decode when setting up a StableDiffusion pipeline --- apps/shark_studio/api/sd.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) 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)