storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-25 08:29:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove pipeline hooks

Browse Source
This commit is contained in:
Aryan
2024-12-23 13:58:37 +01:00
parent 86e1d1888a
commit 5261bed284
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/trainer.py
+4
View File
@@ -814,6 +814,10 @@ class Trainer:
generator=self.state.generator, generator=self.state.generator,
) )
# Remove all hooks that might have been added during pipeline initialization to the models
pipeline.remove_all_hooks()
del pipeline
prompt_filename = string_to_filename(prompt)[:25] prompt_filename = string_to_filename(prompt)[:25]
artifacts = { artifacts = {
"image": {"type": "image", "value": image}, "image": {"type": "image", "value": image},