ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-29 18:51:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed missed merge conflice and locked diffusers version

Browse Source
This commit is contained in:
Jaret Burkett
2025-02-12 09:40:02 -07:00
parent 0b8a32def7
commit 8450aca10e
2 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
jobs/process/BaseSDTrainProcess.py
View File

@@ -1461,19 +1461,13 @@ class BaseSDTrainProcess(BaseTrainProcess):
# print_acc("sage attention is not installed. Using SDP instead")
if self.train_config.gradient_checkpointing:
<<<<<<< HEAD
=======
# if has method enable_gradient_checkpointing
>>>>>>> lumina2
if hasattr(unet, 'enable_gradient_checkpointing'):
unet.enable_gradient_checkpointing()
elif hasattr(unet, 'gradient_checkpointing'):
unet.gradient_checkpointing = True
<<<<<<< HEAD
=======
else:
print("Gradient checkpointing not supported on this model")
>>>>>>> lumina2
if isinstance(text_encoder, list):
for te in text_encoder:
if hasattr(te, 'enable_gradient_checkpointing'):