Add files

app.py

@@ -252,7 +252,7 @@ def t2i_core(model, xt, emb, nemb, step=30, cfg_scale=7.5, return_list=False):
 ########
 
 class wrapper(object):
-    @spaces.GPU()
+    
     def __init__(self, 
                  fp16=False, 
                  tag_diffuser=None, 
@@ -277,13 +277,13 @@ 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)
         return tag_diffuser, tag_lora, tag_scheduler
 
-    @spaces.GPU()
+    
     def load_diffuser_lora(self, tag_diffuser, tag_lora):
         self.net = StableDiffusionPipeline.from_pretrained(
             choices.diffuser[tag_diffuser], torch_dtype=self.torch_dtype).to(self.device)
@@ -295,7 +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
@@ -313,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)
@@ -952,7 +952,7 @@ if __name__ == '__main__':
     css = css_version_4_11_0
 
     wrapper_obj = wrapper(
-        fp16=True, 
+        fp16=False, 
         tag_diffuser=default.diffuser,
         tag_lora=default.lora,
         tag_scheduler=default.scheduler)