mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
major refactoring of modules
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -5,7 +5,7 @@ import random
|
||||
import threading
|
||||
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
||||
import gradio as gr
|
||||
from modules import shared, scripts, devices, processing
|
||||
from modules import shared, scripts_manager, devices, processing
|
||||
|
||||
|
||||
repo_id = "gokaygokay/Flux-Prompt-Enhance"
|
||||
@@ -13,7 +13,7 @@ num_return_sequences = 5
|
||||
load_lock = threading.Lock()
|
||||
|
||||
|
||||
class Script(scripts.Script):
|
||||
class Script(scripts_manager.Script):
|
||||
prompts = [['']]
|
||||
tokenizer: AutoTokenizer = None
|
||||
model: AutoModelForSeq2SeqLM = None
|
||||
|
||||
Reference in New Issue
Block a user