elon example

handler.py

@@ -82,11 +82,11 @@ class EndpointHandler():
             self.pipe.scheduler.config
         )
         
-        scheduler_class_name = "EulerDiscreteScheduler"
-        
-        add_kwargs = {}
-        scheduler = getattr(diffusers, scheduler_class_name)
-        self.pipe.scheduler = scheduler.from_config(self.pipe.scheduler.config, **add_kwargs)
+        # if we need more parameters
+        # scheduler_class_name = "EulerDiscreteScheduler"        
+        # add_kwargs = {}
+        # scheduler = getattr(diffusers, scheduler_class_name)
+        # self.pipe.scheduler = scheduler.from_config(self.pipe.scheduler.config, **add_kwargs)
 
         controlnet_pose_model = "thibaud/controlnet-openpose-sdxl-1.0"
         controlnet_canny_model = "diffusers/controlnet-canny-sdxl-1.0"
@@ -98,6 +98,7 @@ class EndpointHandler():
         controlnet_canny = ControlNetModel.from_pretrained(
             controlnet_canny_model, torch_dtype=dtype
         ).to(device)
+        
         # controlnet_depth = ControlNetModel.from_pretrained(
         #     controlnet_depth_model, torch_dtype=dtype
         # ).to(device)
@@ -144,8 +145,9 @@ class EndpointHandler():
         }
         
         self.app = FaceAnalysis(name="buffalo_l", root="./", providers=["CPUExecutionProvider"])
-        self.app.prepare(ctx_id=0, det_size=(512, 512))
-        self.generator = torch.Generator(device=device.type).manual_seed(3)
+        # con 512 funciona bien
+        self.app.prepare(ctx_id=0, det_size=(640, 640))
+        self.generator = torch.Generator(device=device).manual_seed(42)
 
 
         identitynet_strength_ratio = 0.8
@@ -170,13 +172,14 @@ class EndpointHandler():
         ]
 
     def __call__(self, data):
-        
-        default_prompt = "watercolor painting, {prompt}. vibrant, beautiful, painterly, detailed, textural, artistic"
+
+        default_prompt = "{prompt}, Post-apocalyptic. Mars Colony, Scavengers roam the wastelands searching for valuable resources, rovers, bright morning sunlight shining, (detailed) (intricate) (8k) (HDR) (cinematic lighting) (sharp focus)"
+        # default_prompt = "watercolor painting, {prompt}. vibrant, beautiful, painterly, detailed, textural, artistic"
         default_negative_prompt = "(lowres, low quality, worst quality:1.2), (text:1.2), watermark, anime, photorealistic, 35mm film, deformed, glitch, low contrast, noisy"
 
         # hyperparamters
         # prompt_input = data.pop("inputs", "a man")
-        prompt_input = "a man"
+        prompt_input = "a man flying in the sky in Mars"
         prompt=default_prompt.replace("{prompt}", prompt_input)
         num_inference_steps = data.pop("num_inference_steps", 20)
         guidance_scale = data.pop("guidance_scale", 5.0)
@@ -191,7 +194,7 @@ class EndpointHandler():
         
         # original
         face_image_path = "https://i.ibb.co/GQzm527/examples-musk-resize.jpg"
-        pose_image_path = "https://i.ibb.co/ZSrQ8ZJ/pose.jpg"
+        pose_image_path = "https://i.ibb.co/TRCK4MS/examples-poses-pose2.jpg"
         
         adapter_strength_ratio = 0.8