1580158
/
SD-CN-Animation
mirror of https://github.com/volotat/SD-CN-Animation.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #191 from alexbofa/main 

Update utils.py & txt2vid.py & vid2vid.py
pull/212/head
Alexey Borsky 2023-09-01 09:12:19 +03:00 committed by GitHub
parent 2e257bbfc3 48d8348de4
commit ab1f1583b7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/base_ui.py
View File

@ -6,7 +6,7 @@ from types import SimpleNamespace
from modules import script_callbacks, shared from modules import script_callbacks, shared
from modules.shared import cmd_opts, opts from modules.shared import cmd_opts, opts
from webui import wrap_gradio_gpu_call from modules.call_queue import wrap_gradio_gpu_call
from modules.ui_components import ToolButton, FormRow, FormGroup from modules.ui_components import ToolButton, FormRow, FormGroup
from modules.ui import create_override_settings_dropdown from modules.ui import create_override_settings_dropdown

4
scripts/core/txt2vid.py
View File

@ -6,7 +6,7 @@ import numpy as np
from PIL import Image from PIL import Image
import modules.paths as ph import modules.paths as ph
from modules.shared import devices from modules import devices
from scripts.core import utils, flow_utils from scripts.core import utils, flow_utils
from FloweR.model import FloweR from FloweR.model import FloweR
@ -237,4 +237,4 @@ def start_process(*args):
# print('TOTAL TIME:', int(time.time() - processing_start_time)) # print('TOTAL TIME:', int(time.time() - processing_start_time))
yield 'done', curr_frame, occlusion_mask, warped_styled_frame_, processed_frame, output_video_name, gr.Button.update(interactive=True), gr.Button.update(interactive=False) yield 'done', curr_frame, occlusion_mask, warped_styled_frame_, processed_frame, output_video_name, gr.Button.update(interactive=True), gr.Button.update(interactive=False)

4
scripts/core/utils.py
View File

@ -147,7 +147,7 @@ import modules.processing as processing
from modules.ui import plaintext_to_html from modules.ui import plaintext_to_html
import modules.images as images import modules.images as images
import modules.scripts import modules.scripts
from modules.shared import opts, devices, state from modules.shared import opts, state
from modules import devices, sd_samplers, img2img from modules import devices, sd_samplers, img2img
from modules import shared, sd_hijack, lowvram from modules import shared, sd_hijack, lowvram
@ -429,4 +429,4 @@ def export_settings(*args):
for key in remove_list: for key in remove_list:
if key in args_dict: del args_dict[key] if key in args_dict: del args_dict[key]
return json.dumps(args_dict, indent=2, default=lambda o: getattr(o, '__dict__', str(o))) return json.dumps(args_dict, indent=2, default=lambda o: getattr(o, '__dict__', str(o)))

3
scripts/core/vid2vid.py
View File

@ -11,7 +11,6 @@ from PIL import Image
from modules import devices, sd_samplers from modules import devices, sd_samplers
from modules import shared, sd_hijack, lowvram from modules import shared, sd_hijack, lowvram
from modules.shared import devices
import modules.shared as shared import modules.shared as shared
import gc import gc
@ -267,4 +266,4 @@ def start_process(*args):
warped_styled_frame_ = gr.Image.update() warped_styled_frame_ = gr.Image.update()
processed_frame = gr.Image.update() processed_frame = gr.Image.update()
yield get_cur_stat(), curr_frame, occlusion_mask, warped_styled_frame_, processed_frame, output_video_name, gr.Button.update(interactive=True), gr.Button.update(interactive=False) yield get_cur_stat(), curr_frame, occlusion_mask, warped_styled_frame_, processed_frame, output_video_name, gr.Button.update(interactive=True), gr.Button.update(interactive=False)