Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -5,6 +5,7 @@ from PIL import Image
|
|
5 |
import utils
|
6 |
import datetime
|
7 |
import time
|
|
|
8 |
|
9 |
start_time = time.time()
|
10 |
is_colab = utils.is_google_colab()
|
@@ -98,6 +99,8 @@ def on_model_change(model_name):
|
|
98 |
|
99 |
def inference(model_name, prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt=""):
|
100 |
|
|
|
|
|
101 |
global current_model
|
102 |
for model in models:
|
103 |
if model.name == model_name:
|
|
|
5 |
import utils
|
6 |
import datetime
|
7 |
import time
|
8 |
+
import psutil
|
9 |
|
10 |
start_time = time.time()
|
11 |
is_colab = utils.is_google_colab()
|
|
|
99 |
|
100 |
def inference(model_name, prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt=""):
|
101 |
|
102 |
+
psutil.virtual_memory() # print memory usage
|
103 |
+
|
104 |
global current_model
|
105 |
for model in models:
|
106 |
if model.name == model_name:
|