Dont allow esd-u if less than 40gb of gpu memory

app.py

@@ -103,8 +103,13 @@ class Demo:
                             label="Prompt to Erase",
                             info="Prompt corresponding to concept to erase"
                         )
+
+                        choices = ['ESD-x']
+                        if torch.cuda.get_device_properties(0).total_memory * 1e-9 >= 40:
+                            choices.append('ESD-u')
+                    
                         self.train_method_input = gr.Dropdown(
-                            choices=['ESD-x', 'ESD-u', 'ESD-self'],
+                            choices=choices,
                             value='ESD-x',
                             label='Train Method',
                             info='Method of training'