mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-15 00:37:59 -05:00
68e9281778
* Shark Studio SDXL support, HIP driver support, simpler device info, small fixes * Fixups to llm API/UI and ignore user config files. * Small fixes for unifying pipelines. * Update requirements.txt for iree-turbine (#2130) * Fix Llama2 on CPU (#2133) * Filesystem cleanup and custom model fixes (#2127) * Fix some formatting issues * Remove IREE pin (fixes exe issue) (#2126) * Update find links for IREE packages (#2136) * Shark Studio SDXL support, HIP driver support, simpler device info, small fixes * Abstract out SD pipelines from Studio Webui (WIP) * Switch from pin to minimum torch version and fix index url * Fix device parsing. * Fix linux setup * Fix custom weights. --------- Co-authored-by: saienduri <77521230+saienduri@users.noreply.github.com> Co-authored-by: gpetters-amd <159576198+gpetters-amd@users.noreply.github.com> Co-authored-by: gpetters94 <gpetters@protonmail.com>
76 lines
2.6 KiB
Python
76 lines
2.6 KiB
Python
import os
|
|
import shutil
|
|
from time import time
|
|
|
|
from apps.shark_studio.modules.shared_cmd_opts import cmd_opts
|
|
|
|
shark_tmp = cmd_opts.tmp_dir # os.path.join(os.getcwd(), "shark_tmp/")
|
|
|
|
|
|
def clear_tmp_mlir():
|
|
cleanup_start = time()
|
|
print("Clearing .mlir temporary files from a prior run. This may take some time...")
|
|
mlir_files = [
|
|
filename
|
|
for filename in os.listdir(shark_tmp)
|
|
if os.path.isfile(os.path.join(shark_tmp, filename))
|
|
and filename.endswith(".mlir")
|
|
]
|
|
for filename in mlir_files:
|
|
os.remove(os.path.join(shark_tmp, filename))
|
|
print(f"Clearing .mlir temporary files took {time() - cleanup_start:.4f} seconds.")
|
|
|
|
|
|
def clear_tmp_imgs():
|
|
# tell gradio to use a directory under shark_tmp for its temporary
|
|
# image files unless somewhere else has been set
|
|
if "GRADIO_TEMP_DIR" not in os.environ:
|
|
os.environ["GRADIO_TEMP_DIR"] = os.path.join(shark_tmp, "gradio")
|
|
|
|
print(
|
|
f"gradio temporary image cache located at {os.environ['GRADIO_TEMP_DIR']}. "
|
|
+ "You may change this by setting the GRADIO_TEMP_DIR environment variable."
|
|
)
|
|
|
|
# Clear all gradio tmp images from the last session
|
|
if os.path.exists(os.environ["GRADIO_TEMP_DIR"]):
|
|
cleanup_start = time()
|
|
print(
|
|
"Clearing gradio UI temporary image files from a prior run. This may take some time..."
|
|
)
|
|
shutil.rmtree(os.environ["GRADIO_TEMP_DIR"], ignore_errors=True)
|
|
print(
|
|
f"Clearing gradio UI temporary image files took {time() - cleanup_start:.4f} seconds."
|
|
)
|
|
|
|
# older SHARK versions had to workaround gradio bugs and stored things differently
|
|
else:
|
|
image_files = [
|
|
filename
|
|
for filename in os.listdir(shark_tmp)
|
|
if os.path.isfile(os.path.join(shark_tmp, filename))
|
|
and filename.startswith("tmp")
|
|
and filename.endswith(".png")
|
|
]
|
|
if len(image_files) > 0:
|
|
print(
|
|
"Clearing temporary image files of a prior run of a previous SHARK version. This may take some time..."
|
|
)
|
|
cleanup_start = time()
|
|
for filename in image_files:
|
|
os.remove(shark_tmp + filename)
|
|
print(
|
|
f"Clearing temporary image files took {time() - cleanup_start:.4f} seconds."
|
|
)
|
|
else:
|
|
print("No temporary images files to clear.")
|
|
|
|
|
|
def config_tmp():
|
|
# create shark_tmp if it does not exist
|
|
if not os.path.exists(shark_tmp):
|
|
os.mkdir(shark_tmp)
|
|
|
|
clear_tmp_mlir()
|
|
clear_tmp_imgs()
|