Fix formatting issues. (#903)

This commit is contained in:
Ean Garvey
2023-01-31 21:42:45 -06:00
committed by GitHub
parent 842adef29c
commit a504903dd5
36 changed files with 19 additions and 59 deletions

View File

@@ -88,6 +88,7 @@ txt2img_obj = None
config_obj = None
schedulers = None
# Exposed to UI.
def txt2img_inf(
prompt: str,

View File

@@ -199,7 +199,6 @@ class SharkifyStableDiffusionModel:
return shark_clip
def __call__(self):
for model_id in base_models:
self.inputs = get_input_info(
base_models[model_id],

View File

@@ -114,7 +114,6 @@ class StableDiffusionPipeline:
cpu_scheduling,
return_all_latents=False,
):
step_time_sum = 0
latent_history = [latents]
text_embeddings = torch.from_numpy(text_embeddings).to(dtype)

View File

@@ -1,5 +1,6 @@
from .stable_args import args
# Helper function to profile the vulkan device.
def start_profiling(file_path="foo.rdc", profiling_mode="queue"):
if args.vulkan_debug_utils and "vulkan" in args.device:

View File

@@ -75,7 +75,6 @@ def compile_through_fx(
f16_input_mask=None,
extra_args=[],
):
mlir_module, func_name = import_with_fx(
model, inputs, is_f16, f16_input_mask
)
@@ -89,7 +88,6 @@ def compile_through_fx(
def set_iree_runtime_flags():
vulkan_runtime_flags = [
f"--vulkan_large_heap_block_size={args.vulkan_large_heap_block_size}",
f"--vulkan_validation_layers={'true' if args.vulkan_validation_layers else 'false'}",

View File

@@ -34,7 +34,6 @@ demo_css = resource_path("css/sd_dark_theme.css")
with gr.Blocks(title="Stable Diffusion", css=demo_css) as shark_web:
with gr.Row(elem_id="ui_title"):
nod_logo = Image.open(nodlogo_loc)
logo2 = Image.open(sdlogo_loc)
@@ -55,7 +54,6 @@ with gr.Blocks(title="Stable Diffusion", css=demo_css) as shark_web:
).style(width=150, height=100)
with gr.Row(elem_id="ui_body"):
with gr.Row():
with gr.Column(scale=1, min_width=600):
with gr.Row():