sd-webui/stable-diffusion-webui
1
1
Fork 0
mirror of https://github.com/sd-webui/stable-diffusion-webui.git synced 2024-12-15 07:12:58 +03:00
Code Issues Projects Releases Wiki Activity

Merge pull request #953 from ZeroCool940711/dev

Browse Source 
Fixed an issue with the txt2vid model.
This commit is contained in:
ZeroCool 2022-09-09 19:45:07 -07:00 committed by GitHub
commit 93b6021d38
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
scripts/webui_streamlit.py
View File

@ -1677,16 +1677,15 @@ def txt2vid(
os.makedirs(full_path, exist_ok=True)
# init all of the models and move them to a given GPU
lms = LMSDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear")
#lms = LMSDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear")
try:
if not st.session_state["pipe"]:
st.session_state["pipe"] = StableDiffusionPipeline.from_pretrained(
weights_path,
use_local_file=True,
scheduler=lms,
scheduler=sampler_name,
use_auth_token=True,
torch_dtype=torch.float16,
revision="fp16"
)
@ -1700,9 +1699,8 @@ def txt2vid(
st.session_state["pipe"] = StableDiffusionPipeline.from_pretrained(
weights_path,
use_local_file=True,
scheduler=lms,
scheduler=sampler_name,
use_auth_token=True,
torch_dtype=torch.float16,
revision="fp16"
)
@ -1828,7 +1826,8 @@ def layout():
txt2img_tab, img2img_tab, txt2vid_tab, postprocessing_tab = st.tabs(["Text-to-Image Unified", "Image-to-Image Unified", "Text-to-Video","Post-Processing"])
txt2img_tab, img2img_tab, txt2vid_tab, postprocessing_tab, model_manager_tab = st.tabs(["Text-to-Image Unified", "Image-to-Image Unified",
"Text-to-Video","Post-Processing", "Model Manager"])
with txt2img_tab:
with st.form("txt2img-inputs"):
@ -2249,6 +2248,9 @@ def layout():
# use the current col2 first tab to show the preview_img and update it as its generated.
#preview_image.image(output_images)
with model_manager_tab:
st.write("Model Manager")