Add files

app.py

@@ -277,6 +277,7 @@ class wrapper(object):
         self.cache_inverse_folder = "temp/inverse"
         self.cache_inverse_maxn = 500
 
+    @spaces.GPU()
     def load_all(self, tag_diffuser, tag_lora, tag_scheduler):
         self.load_diffuser_lora(tag_diffuser, tag_lora)
         self.load_scheduler(tag_scheduler)
@@ -294,6 +295,7 @@ class wrapper(object):
         self.tag_lora = tag_lora
         return tag_diffuser, tag_lora
 
+    @spaces.GPU()
     def load_scheduler(self, tag_scheduler):
         self.net.scheduler = choices.scheduler[tag_scheduler].from_config(self.net.scheduler.config)
         self.tag_scheduler = tag_scheduler
@@ -311,7 +313,7 @@ class wrapper(object):
     ##########
     # helper #
     ##########
-
+    @spaces.GPU()
     def precheck_model(self, tag_diffuser, tag_lora, tag_scheduler):
         if (tag_diffuser != self.tag_diffuser) or (tag_lora != self.tag_lora):
             self.load_all(tag_diffuser, tag_lora, tag_scheduler)