up

Author: tolgacangoz

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

@@ -171,7 +171,7 @@ pipeline.scheduler = UniPCMultistepScheduler.from_config(pipeline.scheduler.conf
 
 # Some acceleration helpers
 # Be sure to install Flash Attention: https://github.com/Dao-AILab/flash-attention#installation-and-features
-# Normally 14 min., with compile_repeated_blocks(fullgraph=True) 12 min., with Flash Attention too 5.5 min.
+# Normally 14 min., with compile_repeated_blocks(fullgraph=True) 12 min., with Flash Attention too 5.5 min at A100.
 #pipeline.transformer.set_attention_backend("flash")
 #pipeline.transformer.compile_repeated_blocks(fullgraph=True)
 

src/diffusers/models/transformers/transformer_skyreels_v2.py

@@ -40,7 +40,6 @@
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name
 
 
-# TODO: Copied from doesn't work here?
 def _get_qkv_projections(
     attn: "SkyReelsV2Attention", hidden_states: torch.Tensor, encoder_hidden_states: torch.Tensor
 ):