nalin0503 commited on
Commit
3c73bdd
·
1 Parent(s): ebdd902

check for gpu alloc memory standalone

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -11,6 +11,8 @@ from io import BytesIO
11
  import streamlit as st
12
  from PIL import Image
13
 
 
 
14
  # Set Streamlit page configuration (centered content via CSS)
15
  st.set_page_config(
16
  page_title="Metamorph: DiffMorpher + LCM-LoRA + FILM",
 
11
  import streamlit as st
12
  from PIL import Image
13
 
14
+ print(subprocess.check_output(["nvidia-smi"]).decode())
15
+
16
  # Set Streamlit page configuration (centered content via CSS)
17
  st.set_page_config(
18
  page_title="Metamorph: DiffMorpher + LCM-LoRA + FILM",