anzorq commited on
Commit
2834299
1 Parent(s): 45d8fb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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: