codingggasdfasf
/

video-animator

Model card Files Files and versions

root commited on May 29, 2024

Commit

ef07006

·

1 Parent(s): 96c092e

revertr

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -48,9 +48,9 @@ class EndpointHandler():
         vae = AutoencoderKL.from_pretrained(config_path).to(device, dtype=self.weight_dtype)
         pretrained_base_model_path_unet = os.path.join(base_dir, 'pretrained_weights', 'stable-diffusion-v1-5', 'unet')
         reference_unet = UNet2DConditionModel.from_pretrained(
-            'pretrained_weights/stable-diffusion-v1-5/unet',
             local_files_only=True
         ).to(device, dtype=self.weight_dtype)

         vae = AutoencoderKL.from_pretrained(config_path).to(device, dtype=self.weight_dtype)
         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,
             local_files_only=True
         ).to(device, dtype=self.weight_dtype)