Hugging Face's logo

codingggasdfasf
/
video-animator

Diffusers
ONNX
Model card Files
xet
Community
1
root commited on
Commit
c4b40f2
·
1 Parent(s): fe4e446

reverting

Browse files
Files changed (6) hide show
  1. __pycache__/handler.cpython-310.pyc +0 -0
  2. handler.py +1 -2
  3. input.jpg +0 -0
  4. output.mp4 +0 -0
  5. output/gradio/animation_output.mp4 +0 -0
  6. 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
@@ -50,8 +50,7 @@ class EndpointHandler():
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_weights/stable-diffusion-v1-5/unet',
54
- from_pt=True
55
  ).to(device, dtype=self.weight_dtype)
56
 
57
  inference_config_path = os.path.join(base_dir, 'configs', 'inference', 'inference_v2.yaml')
 
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
  ).to(device, dtype=self.weight_dtype)
55
 
56
  inference_config_path = os.path.join(base_dir, 'configs', 'inference', 'inference_v2.yaml')
input.jpg CHANGED
output.mp4 CHANGED
Binary files a/output.mp4 and b/output.mp4 differ
 
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