Shark Studio SDXL support, HIP driver support, simpler device info, small fixes

This commit is contained in:
Ean Garvey
2024-04-24 04:01:28 -05:00
parent 6d8fb5bbe1
commit f99e794be8

View File

@@ -21,7 +21,6 @@ from apps.shark_studio.web.utils.file_utils import (
get_resource_path,
get_checkpoints_path,
)
from apps.shark_studio.modules.schedulers import get_schedulers
from apps.shark_studio.modules.prompt_encoding import (
get_weighted_text_embeddings,
@@ -56,6 +55,12 @@ EMPTY_FLAGS = {
"pipeline": None,
}
EMPTY_FLAGS = {
"clip": None,
"unet": None,
"vae": None,
"pipeline": None,
}
class StableDiffusion:
# This class is responsible for executing image generation and creating