mirror of https://github.com/vladmandic/automatic
RUF013 update
parent
d3f925e8e5
commit
abe25e7b07
|
|
@ -26,7 +26,7 @@ def auth():
|
|||
return None
|
||||
|
||||
|
||||
def get(endpoint: str, dct: dict = None):
|
||||
def get(endpoint: str, dct: dict | None = None):
|
||||
req = requests.get(f'{sd_url}{endpoint}', json=dct, timeout=300, verify=False, auth=auth())
|
||||
if req.status_code != 200:
|
||||
return { 'error': req.status_code, 'reason': req.reason, 'url': req.url }
|
||||
|
|
@ -34,7 +34,7 @@ def get(endpoint: str, dct: dict = None):
|
|||
return req.json()
|
||||
|
||||
|
||||
def post(endpoint: str, dct: dict = None):
|
||||
def post(endpoint: str, dct: dict | None = None):
|
||||
req = requests.post(f'{sd_url}{endpoint}', json = dct, timeout=None, verify=False, auth=auth())
|
||||
if req.status_code != 200:
|
||||
return { 'error': req.status_code, 'reason': req.reason, 'url': req.url }
|
||||
|
|
|
|||
|
|
@ -30,7 +30,7 @@ def set_progress_bar_config():
|
|||
uni_pc_fm.sample_unipc = sample_unipc
|
||||
|
||||
|
||||
def set_prompt_template(prompt, system_prompt:str=None, optimized_prompt:bool=True, unmodified_prompt:bool=False):
|
||||
def set_prompt_template(prompt, system_prompt: str | None = None, optimized_prompt: bool = True, unmodified_prompt: bool = False):
|
||||
from modules import shared
|
||||
from modules.framepack.pipeline import hunyuan
|
||||
mode = 'unknown'
|
||||
|
|
|
|||
|
|
@ -35,7 +35,7 @@ def split_url(url):
|
|||
return { 'repo': f'{url[0]}/{url[1]}', 'subfolder': url[2] }
|
||||
|
||||
|
||||
def set_model(receipe: str=None):
|
||||
def set_model(receipe: str | None = None):
|
||||
if receipe is None or receipe == '':
|
||||
return
|
||||
lines = [line.strip() for line in receipe.split('\n') if line.strip() != '' and ':' in line]
|
||||
|
|
@ -62,7 +62,7 @@ def reset_model():
|
|||
return ''
|
||||
|
||||
|
||||
def load_model(variant:str=None, pipeline:str=None, text_encoder:str=None, text_encoder_2:str=None, feature_extractor:str=None, image_encoder:str=None, transformer:str=None):
|
||||
def load_model(variant: str | None = None, pipeline: str | None = None, text_encoder: str | None = None, text_encoder_2: str | None = None, feature_extractor: str | None = None, image_encoder: str | None = None, transformer: str | None = None):
|
||||
shared.state.begin('Load FramePack')
|
||||
if variant is not None:
|
||||
if variant not in models.keys():
|
||||
|
|
|
|||
|
|
@ -44,7 +44,7 @@ def worker(
|
|||
mp4_fps, mp4_codec, mp4_sf, mp4_video, mp4_frames, mp4_opt, mp4_ext, mp4_interpolate,
|
||||
vae_type,
|
||||
variant,
|
||||
metadata:dict=None,
|
||||
metadata: dict | None = None,
|
||||
):
|
||||
if metadata is None:
|
||||
metadata = {}
|
||||
|
|
@ -85,7 +85,7 @@ def worker(
|
|||
if not is_f1:
|
||||
prompts = list(reversed(prompts))
|
||||
|
||||
def text_encode(prompt, i:int=None):
|
||||
def text_encode(prompt, i: int | None = None):
|
||||
jobid = shared.state.begin('TE Encode')
|
||||
pbar.update(task, description=f'text encode section={i}')
|
||||
t0 = time.time()
|
||||
|
|
|
|||
Loading…
Reference in New Issue