Add application file

app.py

@@ -61,7 +61,7 @@ class GlobalText:
         self.pipeline              = None
         self.torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
         self.lora_model_state_dict = {}
-        self.device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+        self.device = torch.device("cpu")
 
     def init_source_image_path(self, source_path):
         self.source_paths = sorted(glob(os.path.join(source_path, '*')))
@@ -83,9 +83,9 @@ class GlobalText:
 
         self.scheduler = 'LCM'
         scheduler = LCMScheduler.from_pretrained(model_path, subfolder="scheduler")
-        self.pipeline = ZePoPipeline.from_pretrained(model_path,scheduler=scheduler,torch_dtype=torch.float16,).to('cuda')
-        if is_xformers:
-            self.pipeline.enable_xformers_memory_efficient_attention()
+        self.pipeline = ZePoPipeline.from_pretrained(model_path,scheduler=scheduler,torch_dtype=torch.float16,)
+        # if is_xformers:
+        #     self.pipeline.enable_xformers_memory_efficient_attention()
         time_end = datetime.now()
         print(f'Load {model_path} successful in {time_end-time_start}')
         return gr.Dropdown()