Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
14f6464b
Unverified
Commit
14f6464b
authored
Sep 23, 2024
by
M Saqlain
Committed by
GitHub
Sep 23, 2024
Browse files
[Tests] Reduce the model size in the lumina test (#8985)
* Reduced model size for lumina-tests * Handled failing tests
parent
ba5af5ae
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
6 deletions
+6
-6
tests/pipelines/lumina/test_lumina_nextdit.py
tests/pipelines/lumina/test_lumina_nextdit.py
+6
-6
No files found.
tests/pipelines/lumina/test_lumina_nextdit.py
View file @
14f6464b
...
...
@@ -34,19 +34,19 @@ class LuminaText2ImgPipelinePipelineFastTests(unittest.TestCase, PipelineTesterM
def
get_dummy_components
(
self
):
torch
.
manual_seed
(
0
)
transformer
=
LuminaNextDiT2DModel
(
sample_size
=
16
,
sample_size
=
4
,
patch_size
=
2
,
in_channels
=
4
,
hidden_size
=
2
4
,
hidden_size
=
4
,
num_layers
=
2
,
num_attention_heads
=
3
,
num_attention_heads
=
1
,
num_kv_heads
=
1
,
multiple_of
=
16
,
ffn_dim_multiplier
=
None
,
norm_eps
=
1e-5
,
learn_sigma
=
True
,
qk_norm
=
True
,
cross_attention_dim
=
32
,
cross_attention_dim
=
8
,
scaling_factor
=
1.0
,
)
torch
.
manual_seed
(
0
)
...
...
@@ -57,8 +57,8 @@ class LuminaText2ImgPipelinePipelineFastTests(unittest.TestCase, PipelineTesterM
torch
.
manual_seed
(
0
)
config
=
GemmaConfig
(
head_dim
=
4
,
hidden_size
=
32
,
head_dim
=
2
,
hidden_size
=
8
,
intermediate_size
=
37
,
num_attention_heads
=
4
,
num_hidden_layers
=
2
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment