cocktailpeanut commited on
Commit
eca64de
1 Parent(s): 423769c
Files changed (1) hide show
  1. app2.py +10 -10
app2.py CHANGED
@@ -194,16 +194,16 @@ class AnimateController:
194
  torch.seed()
195
  seed = torch.initial_seed()
196
 
197
- with torch.autocast(device):
198
- sample = pipeline(
199
- prompt_textbox,
200
- negative_prompt=negative_prompt_textbox,
201
- num_inference_steps=sample_step_slider,
202
- guidance_scale=cfg_scale_slider,
203
- width=width_slider,
204
- height=height_slider,
205
- video_length=length_slider,
206
- ).videos
207
 
208
  pipeline.unet.load_state_dict(original_state_dict,strict=False)
209
  del original_state_dict
 
194
  torch.seed()
195
  seed = torch.initial_seed()
196
 
197
+ # with torch.autocast(device):
198
+ sample = pipeline(
199
+ prompt_textbox,
200
+ negative_prompt=negative_prompt_textbox,
201
+ num_inference_steps=sample_step_slider,
202
+ guidance_scale=cfg_scale_slider,
203
+ width=width_slider,
204
+ height=height_slider,
205
+ video_length=length_slider,
206
+ ).videos
207
 
208
  pipeline.unet.load_state_dict(original_state_dict,strict=False)
209
  del original_state_dict