hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-05 03:32:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix NaN issue when running without --precision half

Browse Source
This commit is contained in:
AUTOMATIC1111
2024-06-16 14:09:32 +03:00
parent 80f618ea95
commit 06d0a5ab4d
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modules/models/sd3/sd3_model.py
View File

@@ -149,7 +149,8 @@ class SD3Inferencer(torch.nn.Module):
return contextlib.nullcontext()
def get_learned_conditioning(self, batch: list[str]):
return self.cond_stage_model(batch)
with devices.without_autocast():
return self.cond_stage_model(batch)
def apply_model(self, x, t, cond):
return self.model.apply_model(x, t, c_crossattn=cond['crossattn'], y=cond['vector'])