fix(requirements): update requirements.txt

handler.py

@@ -18,7 +18,6 @@ class EndpointHandler():
         import torch
 
         self.pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
-        print("path", path)
         self.pipe.load_lora_weights("ppierzc/sd-icongenie-model-lora-sdxl", weight_name="pytorch_lora_weights.safetensors")
 
         self.pipe = self.pipe.to(device)
@@ -43,4 +42,4 @@ class EndpointHandler():
         img_str = base64.b64encode(buffered.getvalue())
 
         # postprocess the prediction
-        return {"image": img_str.decode()}
+        return {"image": img_str.decode()}

requirements.txt

@@ -6,3 +6,4 @@ ftfy
 tensorboard
 Jinja2
 peft==0.7.0
+diffusers