VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-18 01:30:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'dev/xy/align_dtype_using_mixed_precision' of github.com:saibit-tech/sd-scripts into dev/xy/align_dtype_using_mixed_precision

Browse Source
This commit is contained in:
Robert
2025-04-22 16:19:52 +08:00
parent c8af252a44 7f984f4775
commit f501209c37
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/flux_models.py
View File

@@ -1005,7 +1005,7 @@ class Flux(nn.Module):
return return
self.offloader_double.prepare_block_devices_before_forward(self.double_blocks) self.offloader_double.prepare_block_devices_before_forward(self.double_blocks)
self.offloader_single.prepare_block_devices_before_forward(self.single_blocks) self.offloader_single.prepare_block_devices_before_forward(self.single_blocks)
def forward( def forward(
self, self,
img: Tensor, img: Tensor,