trying

handler.py

@@ -50,7 +50,8 @@ class EndpointHandler():
         pretrained_base_model_path_unet = os.path.join(base_dir, 'pretrained_weights', 'stable-diffusion-v1-5', 'unet')
         print("model path is " + pretrained_base_model_path_unet)
         reference_unet = UNet2DConditionModel.from_pretrained(
-            pretrained_base_model_path_unet
+            pretrained_base_model_path_unet,
+            use_safetensors = True
         ).to(dtype=self.weight_dtype, device="cuda")
 
         inference_config_path = os.path.join(base_dir, 'configs', 'inference', 'inference_v2.yaml')

requirements.txt

@@ -24,6 +24,10 @@ gradio==3.41.2
 onnxruntime-coreml==1.13.1; python_version == '3.9' and sys_platform == 'darwin' and platform_machine != 'arm64'
 transformers==4.41.1
 
+# Stuff huggingface is complaining about
+einops==0.4.1
+av==11.0.0
+
 # Add additional dependencies
 diffusers==0.24.0
 omegaconf==2.2.3
@@ -33,10 +37,6 @@ safetensors==0.4.0
 facenet-pytorch==2.5.2
 dlib==19.22.0
 
-# Stuff huggingface is complaining about
-einops==0.4.1
-av==11.0.0
-
 # Additional dependencies from the first list not present in the second list
 accelerate==0.21.0
 clip @ https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip#sha256=b5842c25da441d6c581b53a5c60e0c2127ebafe0f746f8e15561a006c6c3be6a