AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

remove guidance

Browse Source
This commit is contained in:
Edna
2025-06-11 20:45:45 -06:00
committed by GitHub
parent e69d73099d
commit 01bc0dcc56
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/diffusers/models/transformers/transformer_chroma.py
View File

@@ -416,7 +416,6 @@ class ChromaTransformer2DModel(
timestep: torch.LongTensor = None,
img_ids: torch.Tensor = None,
txt_ids: torch.Tensor = None,
guidance: torch.Tensor = None,
joint_attention_kwargs: Optional[Dict[str, Any]] = None,
controlnet_block_samples=None,
controlnet_single_block_samples=None,
@@ -465,10 +464,8 @@ class ChromaTransformer2DModel(
hidden_states = self.x_embedder(hidden_states)
timestep = timestep.to(hidden_states.dtype) * 1000
if guidance is not None:
guidance = guidance.to(hidden_states.dtype) * 1000
input_vec = self.time_text_embed(timestep, guidance)
input_vec = self.time_text_embed(timestep)
pooled_temb = self.distilled_guidance_layer(input_vec)
encoder_hidden_states = self.context_embedder(encoder_hidden_states)