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 autocast for text encoder forwad

Browse Source
This commit is contained in:
davidb
2025-10-09 16:27:09 +00:00
committed by DavidBert
parent d71ddd0079
commit 14903ee599
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/diffusers/pipelines/photon/pipeline_photon.py
View File

@@ -381,12 +381,11 @@ class PhotonPipeline(
attention_mask = tokens["attention_mask"].bool().to(device)
with torch.no_grad():
with torch.autocast("cuda", enabled=False):
emb = self.text_encoder(
input_ids=input_ids,
attention_mask=attention_mask,
output_hidden_states=True,
)
emb = self.text_encoder(
input_ids=input_ids,
attention_mask=attention_mask,
output_hidden_states=True,
)
all_embeddings = emb["last_hidden_state"]