Spaces:
Runtime error
Runtime error
Update model.py
Browse files
model.py
CHANGED
|
@@ -216,8 +216,8 @@ class Model:
|
|
| 216 |
elif preprocessor_name == 'No preprocess':
|
| 217 |
condition_img = image
|
| 218 |
print('get depth')
|
| 219 |
-
del self.preprocessor.model
|
| 220 |
-
torch.cuda.empty_cache()
|
| 221 |
condition_img = condition_img.resize((512,512))
|
| 222 |
W, H = condition_img.size
|
| 223 |
|
|
|
|
| 216 |
elif preprocessor_name == 'No preprocess':
|
| 217 |
condition_img = image
|
| 218 |
print('get depth')
|
| 219 |
+
# del self.preprocessor.model
|
| 220 |
+
# torch.cuda.empty_cache()
|
| 221 |
condition_img = condition_img.resize((512,512))
|
| 222 |
W, H = condition_img.size
|
| 223 |
|