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')
 
         reference_unet = UNet2DConditionModel.from_pretrained(
-            pretrained_base_model_path_unet,
+            'pretrained_weights/stable-diffusion-v1-5/unet',
+            local_files_only=True
         ).to(device, dtype=self.weight_dtype)
 
         inference_config_path = os.path.join(base_dir, 'configs', 'inference', 'inference_v2.yaml')