jbilcke-hf HF staff commited on
Commit
f4b832e
·
verified ·
1 Parent(s): 222306d

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +5 -0
handler.py CHANGED
@@ -12,9 +12,14 @@ import traceback
12
  import os
13
  import numpy as np
14
  import torch
 
 
 
 
15
  from diffusers import LTXPipeline, LTXImageToVideoPipeline
16
  from diffusers.pipelines.teacache_utils import apply_teacache
17
  from diffusers.hooks import apply_enhance_a_video, EnhanceAVideoConfig
 
18
  from PIL import Image
19
 
20
  from varnish import Varnish
 
12
  import os
13
  import numpy as np
14
  import torch
15
+
16
+ import monkey
17
+ monkey.apply_patches()
18
+
19
  from diffusers import LTXPipeline, LTXImageToVideoPipeline
20
  from diffusers.pipelines.teacache_utils import apply_teacache
21
  from diffusers.hooks import apply_enhance_a_video, EnhanceAVideoConfig
22
+
23
  from PIL import Image
24
 
25
  from varnish import Varnish