diff --git a/modules/framepack/create-video.py b/modules/framepack/create-video.py index f1b334d3a..1863a2e72 100755 --- a/modules/framepack/create-video.py +++ b/modules/framepack/create-video.py @@ -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 } diff --git a/modules/framepack/framepack_hijack.py b/modules/framepack/framepack_hijack.py index 4ce623827..63f4b24f0 100644 --- a/modules/framepack/framepack_hijack.py +++ b/modules/framepack/framepack_hijack.py @@ -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' diff --git a/modules/framepack/framepack_load.py b/modules/framepack/framepack_load.py index 21f0819ce..74e853a52 100644 --- a/modules/framepack/framepack_load.py +++ b/modules/framepack/framepack_load.py @@ -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(): diff --git a/modules/framepack/framepack_worker.py b/modules/framepack/framepack_worker.py index 85f87a624..fdd9637c9 100644 --- a/modules/framepack/framepack_worker.py +++ b/modules/framepack/framepack_worker.py @@ -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()