mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 01:04:32 +02:00
major refactoring of modules
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -2,7 +2,7 @@ import os
|
||||
import gradio as gr
|
||||
import diffusers
|
||||
from safetensors.torch import load_file
|
||||
from modules import scripts, processing, shared, devices, sd_models
|
||||
from modules import scripts_manager, processing, shared, devices, sd_models
|
||||
|
||||
|
||||
# config
|
||||
@@ -189,12 +189,12 @@ def set_free_noise(frames):
|
||||
shared.sd_model.enable_free_noise(context_length=context_length, context_stride=context_stride)
|
||||
|
||||
|
||||
class Script(scripts.Script):
|
||||
class Script(scripts_manager.Script):
|
||||
def title(self):
|
||||
return 'Video: AnimateDiff'
|
||||
|
||||
def show(self, is_img2img):
|
||||
# return scripts.AlwaysVisible if shared.native else False
|
||||
# return scripts_manager.AlwaysVisible if shared.native else False
|
||||
return not is_img2img
|
||||
|
||||
|
||||
@@ -231,7 +231,7 @@ class Script(scripts.Script):
|
||||
lora = LORAS[lora_index]
|
||||
set_adapter(adapter)
|
||||
if motion_adapter is None:
|
||||
return
|
||||
return None
|
||||
set_scheduler(p, adapter, override_scheduler)
|
||||
set_lora(p, lora, strength)
|
||||
set_free_init(fi_method, fi_iters, fi_order, fi_spatial, fi_temporal)
|
||||
|
||||
Reference in New Issue
Block a user