Max Reimann
commited on
Commit
•
c7937b8
1
Parent(s):
9109365
clear cuda mem after optimize
Browse files- worker/serve.py +2 -0
worker/serve.py
CHANGED
|
@@ -160,6 +160,8 @@ class StyleTask:
|
|
| 160 |
return
|
| 161 |
|
| 162 |
self.status = "finished"
|
|
|
|
|
|
|
| 163 |
print("finished styling task: " + str(self.task_id))
|
| 164 |
|
| 165 |
class StylerQueue:
|
|
|
|
| 160 |
return
|
| 161 |
|
| 162 |
self.status = "finished"
|
| 163 |
+
del self.neural_optimizer
|
| 164 |
+
torch.cuda.empty_cache()
|
| 165 |
print("finished styling task: " + str(self.task_id))
|
| 166 |
|
| 167 |
class StylerQueue:
|