wren93 commited on
Commit
defcb9b
·
1 Parent(s): 7196d13
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ class AnimateController:
44
 
45
  self.image_resolution = (256, 256)
46
  # config models
47
- self.pipeline = ConditionalAnimationPipeline.from_pretrained("TIGER-Lab/ConsistI2V", torch_dtype=torch.float16,)
48
  self.pipeline.to("cuda")
49
 
50
  def update_textbox_and_save_image(self, input_image, height_slider, width_slider, center_crop):
 
44
 
45
  self.image_resolution = (256, 256)
46
  # config models
47
+ self.pipeline = ConditionalAnimationPipeline.from_pretrained("TIGER-Lab/ConsistI2V")
48
  self.pipeline.to("cuda")
49
 
50
  def update_textbox_and_save_image(self, input_image, height_slider, width_slider, center_crop):