We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e6f5f4b commit 7517b9fCopy full SHA for 7517b9f
1 file changed
README.md
@@ -51,7 +51,7 @@ from audio_diffusion_pytorch import DiffusionModel, UNetV0, VDiffusion, VSampler
51
model = DiffusionModel(
52
# ... same as unconditional model
53
use_text_conditioning=True, # U-Net: enables text conditioning (default T5-base)
54
- use_embedding_cfg=True # U-Net: enables classifier free guidance
+ use_embedding_cfg=True, # U-Net: enables classifier free guidance
55
embedding_max_length=64, # U-Net: text embedding maximum length (default for T5-base)
56
embedding_features=768, # U-Net: text mbedding features (default for T5-base)
57
cross_attentions=[0, 0, 0, 1, 1, 1, 1, 1, 1], # U-Net: cross-attention enabled/disabled at each layer
0 commit comments