Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
|
@@ -71,6 +71,8 @@ def infer(
|
|
| 71 |
out_str += tmp
|
| 72 |
yield out_str.strip()
|
| 73 |
out_last = i + 1
|
|
|
|
|
|
|
| 74 |
yield out_str.strip()
|
| 75 |
|
| 76 |
examples = [
|
|
|
|
| 71 |
out_str += tmp
|
| 72 |
yield out_str.strip()
|
| 73 |
out_last = i + 1
|
| 74 |
+
gc.collect()
|
| 75 |
+
torch.cuda.empty_cache()
|
| 76 |
yield out_str.strip()
|
| 77 |
|
| 78 |
examples = [
|