Spaces:

baulab
/

Erasing-Concepts-In-Diffusion

Runtime error

RohitGandikota commited on Mar 25, 2023

Commit

7bf73ed

1 Parent(s): 079d713

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from tqdm import tqdm
 from train import train
 import os
-os.system("nvidia-smi -L")
 model_map = {'Van Gogh' : 'models/vangogh.pt',
              'Pablo Picasso': 'models/pablopicasso.pt',

 from train import train
 import os
+os.system("nvidia-smi topo -m")
 model_map = {'Van Gogh' : 'models/vangogh.pt',
              'Pablo Picasso': 'models/pablopicasso.pt',