Added save_video_on_stop option to the Settings page, this option is …

…used to save the txt2vid generated video to disk when we hit the stop button mid generation.
Sygil-Dev · Oct 8, 2022 · fe6e72f · fe6e72f
1 parent 63b2ff2
commit fe6e72f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -34,6 +34,7 @@ streamlit-tensorboard==0.0.2
 hydralit==1.0.14
 hydralit_components==1.0.10
 stqdm==0.0.4
+diffusers==0.4.1
 
 # Img2text
 ftfy==6.1.1
@@ -51,7 +52,6 @@ piexif==1.1.3  # used by sd_utils
 
 accelerate==0.12.0
 albumentations==0.4.3
-diffusers==0.3.0
 einops==0.3.1
 facexlib>=0.2.3
 imageio-ffmpeg==0.4.2

diff --git a/scripts/Settings.py b/scripts/Settings.py
@@ -685,6 +685,10 @@ def layout():
 				st.session_state["defaults"].txt2vid.save_video = st.checkbox("Save Txt2Vid Video", value=st.session_state['defaults'].txt2vid.save_video,
 																			  help="Choose to save the Txt2Vid video. Default: True")
 
+				st.session_state["defaults"].txt2vid.save_video_on_stop = st.checkbox("Save video on Stop",value=st.session_state['defaults'].txt2vid.save_video_on_stop,
+																					  help="Save a video with all the images generated as frames when we hit the stop button \
+																					  during a generation.")
+
 				st.session_state["defaults"].txt2vid.group_by_prompt = st.checkbox("Group By txt2vid Prompt", value=st.session_state['defaults'].txt2vid.group_by_prompt,
 																				   help="Choose to save images grouped by their prompt. Default: False")