Spaces:
Runtime error
Runtime error
Commit
•
fd9afda
1
Parent(s):
75cb83b
Update train.py
Browse files
train.py
CHANGED
|
@@ -66,7 +66,8 @@ def train(prompt, modules, freeze_modules, iterations, negative_guidance, lr, sa
|
|
| 66 |
optimizer.step()
|
| 67 |
|
| 68 |
torch.save(finetuner.state_dict(), save_path)
|
| 69 |
-
|
|
|
|
| 70 |
if __name__ == '__main__':
|
| 71 |
|
| 72 |
import argparse
|
|
|
|
| 66 |
optimizer.step()
|
| 67 |
|
| 68 |
torch.save(finetuner.state_dict(), save_path)
|
| 69 |
+
del diffuser
|
| 70 |
+
torch.cuda.empty_cache()
|
| 71 |
if __name__ == '__main__':
|
| 72 |
|
| 73 |
import argparse
|