monster/Qwen-Image-Layered-MRP-MLX
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix guidance scale

Browse Source
This commit is contained in:
Fabio 2024-08-18 17:56:31 +02:00
parent 6977e92061
commit acfc448f0c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/flux_1_schnell/models/transformer/transformer.py
View File

@ -37,7 +37,7 @@ class Transformer(nn.Module):
time_step = config.sigmas[t] * config.num_train_steps time_step = config.sigmas[t] * config.num_train_steps
time_step = mx.broadcast_to(time_step, (1,)) time_step = mx.broadcast_to(time_step, (1,))
hidden_states = self.x_embedder(hidden_states) hidden_states = self.x_embedder(hidden_states)
guidance = mx.broadcast_to(config.guidance, (1,)) guidance = mx.broadcast_to(config.guidance * config.num_train_steps, (1,))
text_embeddings = self.time_text_embed.forward(time_step, pooled_prompt_embeds, guidance) text_embeddings = self.time_text_embed.forward(time_step, pooled_prompt_embeds, guidance)
encoder_hidden_states = self.context_embedder(prompt_embeds) encoder_hidden_states = self.context_embedder(prompt_embeds)
txt_ids = Transformer._prepare_text_ids(seq_len = prompt_embeds.shape[1]) txt_ids = Transformer._prepare_text_ids(seq_len = prompt_embeds.shape[1])