iitolstykh commited on
Commit
3e983ee
·
verified ·
1 Parent(s): eb3d4ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -43,6 +43,8 @@ def load_pipeline():
43
  device="cuda",
44
  )
45
 
 
 
46
  return editor_pipeline
47
 
48
  def set_env(seed=0):
 
43
  device="cuda",
44
  )
45
 
46
+ print(f"Model loaded. Model device: {editor_pipeline.pipe.device}")
47
+
48
  return editor_pipeline
49
 
50
  def set_env(seed=0):