Update pipline_StableDiffusion_ConsistentID.py

pipline_StableDiffusion_ConsistentID.py

@@ -64,7 +64,7 @@ class ConsistentIDStableDiffusionPipeline(StableDiffusionPipeline):
 
         ### BiSeNet
         self.bise_net = BiSeNet(n_classes = 19)
-        self.bise_net.cuda()
+        # self.bise_net.cuda() # CUDA must not be initialized in the main process on Spaces with Stateless GPU environment
         self.bise_net_cp=bise_net_cp_path
         self.bise_net.load_state_dict(torch.load(self.bise_net_cp))
         self.bise_net.eval()