Update pipline_StableDiffusion_ConsistentID.py

pipline_StableDiffusion_ConsistentID.py

@@ -66,7 +66,7 @@ class ConsistentIDStableDiffusionPipeline(StableDiffusionPipeline):
         self.bise_net = BiSeNet(n_classes = 19)
         # 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.load_state_dict(torch.load(self.bise_net_cp), map_location=torch.device('cpu'))
         self.bise_net.eval()
         # Colors for all 20 parts
         self.part_colors = [[255, 0, 0], [255, 85, 0], [255, 170, 0],