Hugging Face's logo

codingggasdfasf
/
video-animator

Diffusers
ONNX
Model card Files
xet
Community
1
root commited on
Commit
f751834
·
1 Parent(s): 7083a27

unet path

Browse files
Files changed (5) hide show
  1. __pycache__/handler.cpython-310.pyc +0 -0
  2. handler.py +2 -2
  3. input.jpg +0 -0
  4. output/gradio/animation_output.mp4 +0 -0
  5. output/gradio/cropped_face.jpg +0 -0
__pycache__/handler.cpython-310.pyc CHANGED
Binary files a/__pycache__/handler.cpython-310.pyc and b/__pycache__/handler.cpython-310.pyc differ
 
handler.py CHANGED
@@ -49,9 +49,9 @@ class EndpointHandler():
49
 
50
  pretrained_base_model_path_unet = os.path.join(base_dir, 'pretrained_weights', 'stable-diffusion-v1-5', 'unet')
51
 
 
52
  reference_unet = UNet2DConditionModel.from_pretrained(
53
- pretrained_base_model_path_unet,
54
- variant="fp16",
55
  ).to(device, dtype=self.weight_dtype)
56
 
57
  inference_config_path = os.path.join(base_dir, 'configs', 'inference', 'inference_v2.yaml')
 
49
 
50
  pretrained_base_model_path_unet = os.path.join(base_dir, 'pretrained_weights', 'stable-diffusion-v1-5', 'unet')
51
 
52
+ print("model path is " + pretrained_base_model_path_unet)
53
  reference_unet = UNet2DConditionModel.from_pretrained(
54
+ "./pretrained_weights/stable-diffusion-v1-5/unet",
 
55
  ).to(device, dtype=self.weight_dtype)
56
 
57
  inference_config_path = os.path.join(base_dir, 'configs', 'inference', 'inference_v2.yaml')
input.jpg CHANGED
output/gradio/animation_output.mp4 CHANGED
Binary files a/output/gradio/animation_output.mp4 and b/output/gradio/animation_output.mp4 differ
 
output/gradio/cropped_face.jpg CHANGED