Update src/flux/generate.py
Browse files- src/flux/generate.py +1 -1
src/flux/generate.py
CHANGED
@@ -708,7 +708,7 @@ def generate_from_test_sample(
|
|
708 |
return delta_emb, delta_emb_pblock, delta_emb_mask, \
|
709 |
text_cond_mask, delta_start_ends, condition_latents, condition_ids
|
710 |
|
711 |
-
num_inference_steps =
|
712 |
num_channels_latents = pipe.transformer.config.in_channels // 4
|
713 |
|
714 |
# set timesteps
|
|
|
708 |
return delta_emb, delta_emb_pblock, delta_emb_mask, \
|
709 |
text_cond_mask, delta_start_ends, condition_latents, condition_ids
|
710 |
|
711 |
+
num_inference_steps = 28 # FIXME: harcoded here
|
712 |
num_channels_latents = pipe.transformer.config.in_channels // 4
|
713 |
|
714 |
# set timesteps
|