MCPcopy Create free account

hub / github.com/SkyworkAI/SkyReels-V1 / functions

Functions42 in github.com/SkyworkAI/SkyReels-V1

⨍ Functions 42◇ Types & classes 7
↓ 4 callersMethodempty_cache_if_needed
(self)
skyreelsinfer/offload.py:417
↓ 3 callersMethodgpu_unload_blocks
(self, model_name, blocks_name)
skyreelsinfer/offload.py:325
↓ 3 callersMethodready_to_check_mem
(self)
skyreelsinfer/offload.py:407
↓ 2 callersMethodgpu_load_blocks
(self, model_name, blocks_name, async_load=False)
skyreelsinfer/offload.py:282
↓ 2 callersMethodinference
(self, kwargs: Dict[str, Any])
skyreelsinfer/skyreels_video_infer.py:254
↓ 1 callersMethod_load_model
( self, model_id: str, base_model_id: str = "hunyuanvideo-community/HunyuanVideo",
skyreelsinfer/skyreels_video_infer.py:35
↓ 1 callersMethodadd_module_to_blocks
(self, model_name, blocks_name, submodule, prev_block_name)
skyreelsinfer/offload.py:236
↓ 1 callersMethodany_param_or_buffer
(self, target_module: torch.nn.Module)
skyreelsinfer/offload.py:428
↓ 1 callersMethodcan_model_be_cotenant
(self, model_name)
skyreelsinfer/offload.py:268
↓ 1 callersFunctioncreate_gradio_interface
Create a Gradio interface based on the task type.
scripts/gradio_web.py:69
↓ 1 callersMethoddamon_inference
(self, request_queue: mp.Queue, response_queue: mp.Queue)
skyreelsinfer/skyreels_video_infer.py:152
↓ 1 callersMethodencode_prompt
( self, prompt: Union[str, List[str]], do_classifier_free_guidance: bool, nega
skyreelsinfer/pipelines/pipeline_skyreels_video.py:68
↓ 1 callersFunctionget_transformer_model_id
(task_type:str)
scripts/gradio_web.py:17
↓ 1 callersMethodgpu_load
(self, model_name)
skyreelsinfer/offload.py:356
↓ 1 callersMethodhook_me
(self, target_module, model, model_name, module_id, previous_method)
skyreelsinfer/offload.py:486
↓ 1 callersMethodhook_me_light
(self, target_module, model_name, blocks_name, previous_method, context)
skyreelsinfer/offload.py:438
↓ 1 callersMethodimage_latents
( self, initial_image, batch_size, height, width, device,
skyreelsinfer/pipelines/pipeline_skyreels_video.py:132
↓ 1 callersFunctioninit_predictor
(task_type:str, gpu_num:int=1)
scripts/gradio_web.py:20
↓ 1 callersMethodmove_args_to_gpu
(self, *args, **kwargs)
skyreelsinfer/offload.py:385
↓ 1 callersMethodoffload
Enable offloading for multiple models in the pipeline, supporting video generation inference on user-level GPUs. pipe: the pipeline o
skyreelsinfer/offload.py:57
↓ 1 callersFunctionrescale_noise_cfg
Rescale `noise_cfg` according to `guidance_rescale`. Based on findings of [Common Diffusion Noise Schedules and Sample Steps are Flawed](http
skyreelsinfer/pipelines/pipeline_skyreels_video.py:35
↓ 1 callersFunctionresizecrop
(image, th, tw)
skyreelsinfer/pipelines/pipeline_skyreels_video.py:19
↓ 1 callersMethodunload_all
(self, model_name: str)
skyreelsinfer/offload.py:366
↓ 1 callersMethodwarm_up
(self)
skyreelsinfer/skyreels_video_infer.py:136
Method__call__
( self, prompt: str, negative_prompt: str = "Aerial view, aerial view, overexposed, lo
skyreelsinfer/pipelines/pipeline_skyreels_video.py:161
Method__init__
( self, task_type: TaskType, model_id: str, quant_model: bool = True,
skyreelsinfer/skyreels_video_infer.py:71
Method__init__
( self, task_type: TaskType, model_id: str, quant_model: bool = True,
skyreelsinfer/skyreels_video_infer.py:196
Method__init__
(self)
skyreelsinfer/offload.py:24
Method__init__
(self)
skyreelsinfer/offload.py:33
Methodcheck_change_module
(module, *args, **kwargs)
skyreelsinfer/offload.py:487
Methodcheck_empty_cuda_cache
(module, *args, **kwargs)
skyreelsinfer/offload.py:442
Methodclip_skip
(self)
skyreelsinfer/pipelines/pipeline_skyreels_video.py:60
Methodcpu_to_gpu
(stream_to_use, blocks_params, record_for_stream=None)
skyreelsinfer/offload.py:286
Methoddetach_hook
(self, module)
skyreelsinfer/offload.py:28
Methoddo_classifier_free_guidance
(self)
skyreelsinfer/pipelines/pipeline_skyreels_video.py:64
Functiongenerate_video
(prompt, seed, image=None)
scripts/gradio_web.py:35
Methodguidance_rescale
(self)
skyreelsinfer/pipelines/pipeline_skyreels_video.py:56
Methodlauch_single_gpu_infer
( self, task_type: TaskType, model_id: str, quant_model: bool = True,
skyreelsinfer/skyreels_video_infer.py:225
Methodload_module_blocks
(module, *args, **kwargs)
skyreelsinfer/offload.py:447
Methodmove_text_encoder_to_cpu
(pipe)
skyreelsinfer/offload.py:100
Methodmove_text_encoder_to_gpu
(pipe)
skyreelsinfer/offload.py:95
Functionsingle_gpu_run
( rank, task_type: TaskType, model_id: str, request_queue: mp.Queue, response_queue: mp.Qu
skyreelsinfer/skyreels_video_infer.py:170