Spaces:
Paused
Paused
Commit
·
dec3828
1
Parent(s):
34f173c
Clear cache attempt
Browse files
app.py
CHANGED
|
@@ -42,6 +42,7 @@ from random import randint
|
|
| 42 |
from pathvalidate import sanitize_filename
|
| 43 |
from huggingface_hub import hf_hub_download
|
| 44 |
import shortuuid
|
|
|
|
| 45 |
|
| 46 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
| 47 |
print("Using device:", device)
|
|
@@ -76,7 +77,12 @@ perceptor = (
|
|
| 76 |
.requires_grad_(False)
|
| 77 |
.to(device)
|
| 78 |
)
|
|
|
|
|
|
|
|
|
|
| 79 |
def run_all(user_input, width, height, template, num_steps, flavor):
|
|
|
|
|
|
|
| 80 |
import random
|
| 81 |
#if uploaded_file is not None:
|
| 82 |
#uploaded_folder = f"{DefaultPaths.root_path}/uploaded"
|
|
|
|
| 42 |
from pathvalidate import sanitize_filename
|
| 43 |
from huggingface_hub import hf_hub_download
|
| 44 |
import shortuuid
|
| 45 |
+
import gc
|
| 46 |
|
| 47 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
| 48 |
print("Using device:", device)
|
|
|
|
| 77 |
.requires_grad_(False)
|
| 78 |
.to(device)
|
| 79 |
)
|
| 80 |
+
gc.collect()
|
| 81 |
+
torch.cuda.empty_cache()
|
| 82 |
+
|
| 83 |
def run_all(user_input, width, height, template, num_steps, flavor):
|
| 84 |
+
gc.collect()
|
| 85 |
+
torch.cuda.empty_cache()
|
| 86 |
import random
|
| 87 |
#if uploaded_file is not None:
|
| 88 |
#uploaded_folder = f"{DefaultPaths.root_path}/uploaded"
|