Skip to content

Commit df1f6b7

Browse files
authored
Clean up comments in skyreels_v2.md
Removed comments about acceleration helpers and Flash Attention installation.
1 parent 14240d2 commit df1f6b7

File tree

1 file changed

+1
-10
lines changed

1 file changed

+1
-10
lines changed

docs/source/en/api/pipelines/skyreels_v2.md

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -182,15 +182,6 @@ pipeline.to("cuda")
182182
flow_shift = 8.0 # 8.0 for T2V, 5.0 for I2V
183183
pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.config, flow_shift=flow_shift)
184184

185-
# Some acceleration helpers
186-
# Be sure to install Flash Attention: https://github.com/Dao-AILab/flash-attention#installation-and-features
187-
# Normally 14 min., with compile_repeated_blocks(fullgraph=True) 12 min., with Flash Attention too less min. at A100.
188-
# If you want to follow the original implementation in terms of attentions:
189-
#for block in pipeline.transformer.blocks:
190-
# block.attn1.set_attention_backend("_native_cudnn")
191-
# block.attn2.set_attention_backend("flash_varlen") # or "_flash_varlen_3"
192-
#pipeline.transformer.compile_repeated_blocks(fullgraph=True)
193-
194185
prompt = "A cat and a dog baking a cake together in a kitchen. The cat is carefully measuring flour, while the dog is stirring the batter with a wooden spoon. The kitchen is cozy, with sunlight streaming through the window."
195186

196187
output = pipeline(
@@ -352,4 +343,4 @@ export_to_video(output, "video.mp4", fps=24, quality=8)
352343

353344
## SkyReelsV2PipelineOutput
354345

355-
[[autodoc]] pipelines.skyreels_v2.pipeline_output.SkyReelsV2PipelineOutput
346+
[[autodoc]] pipelines.skyreels_v2.pipeline_output.SkyReelsV2PipelineOutput

0 commit comments

Comments
 (0)