pull in import fix

This commit is contained in:
Stephan Auerhahn
2023-08-09 13:29:16 -07:00
parent a009aa8a9f
commit 725bea9f75

View File

@@ -1,7 +1,13 @@
from dataclasses import asdict
import os
import numpy as np
import streamlit as st
import torch
from einops import repeat
from pytorch_lightning import seed_everything
from sgm.inference.api import (
SamplingSpec,
SamplingParams,
ModelArchitecture,
SamplingPipeline,
@@ -11,7 +17,17 @@ from sgm.inference.helpers import (
get_unique_embedder_keys_from_conditioner,
perform_save_locally,
)
from scripts.demo.streamlit_helpers import *
from scripts.demo.streamlit_helpers import (
get_interactive_image,
get_unique_embedder_keys_from_conditioner,
init_embedder_options,
init_sampling,
init_save_locally,
init_st,
perform_save_locally,
set_lowvram_mode,
show_samples,
)
SAVE_PATH = "outputs/demo/txt2img/"