Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -94,6 +94,8 @@ print('Loading model checkpoint...')
|
|
| 94 |
|
| 95 |
model.load_state_dict(torch.load(MODEL_CHECKPOINT, map_location='cpu'))
|
| 96 |
|
|
|
|
|
|
|
| 97 |
print('=' * 70)
|
| 98 |
print('Done!')
|
| 99 |
print('=' * 70)
|
|
|
|
| 94 |
|
| 95 |
model.load_state_dict(torch.load(MODEL_CHECKPOINT, map_location='cpu'))
|
| 96 |
|
| 97 |
+
model = torch.compile(model, mode='max-autotune')
|
| 98 |
+
|
| 99 |
print('=' * 70)
|
| 100 |
print('Done!')
|
| 101 |
print('=' * 70)
|