mirror of
https://github.com/Stability-AI/generative-models.git
synced 2026-02-02 04:14:27 +01:00
pull in import fix
This commit is contained in:
@@ -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/"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user