Skip to content

Commit 9871c7d

Browse files
jfacevedo-googlejfacevedo-google
committed
removed unused code.
1 parent 4a12b39 commit 9871c7d

5 files changed

Lines changed: 4 additions & 623 deletions

File tree

‎src/maxdiffusion/configs/base_flux_dev.yml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,7 @@ logical_axis_rules: [
130130
['activation_batch', ['data','fsdp']],
131131
['activation_heads', 'tensor'],
132132
['activation_kv', 'tensor'],
133+
['mlp','tensor'],
133134
['embed','fsdp'],
134135
['heads', 'tensor'],
135136
['conv_batch', ['data','fsdp']],

‎src/maxdiffusion/configs/base_flux_schnell.yml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,7 @@ logical_axis_rules: [
129129
['activation_batch', ['data','fsdp']],
130130
['activation_heads', 'tensor'],
131131
['activation_kv', 'tensor'],
132+
['mlp','tensor'],
132133
['embed','fsdp'],
133134
['heads', 'tensor'],
134135
['conv_batch', ['data','fsdp']],

‎src/maxdiffusion/models/flux/modules/__init__.py‎

Lines changed: 0 additions & 15 deletions
This file was deleted.

0 commit comments

Comments
 (0)