kanishka089 commited on
Commit
c46a063
·
verified ·
1 Parent(s): 5d7b556

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def generate_gif(custom_prompt):
38
  num_frames=32,
39
  guidance_scale=2.0,
40
  num_inference_steps=6,
41
- generator=torch.Generator("cuda").manual_seed(0),
42
  )
43
  frames = output.frames[0]
44
  export_to_gif(frames, "animatelcm.gif")
 
38
  num_frames=32,
39
  guidance_scale=2.0,
40
  num_inference_steps=6,
41
+ generator=torch.Generator("cpu").manual_seed(0),
42
  )
43
  frames = output.frames[0]
44
  export_to_gif(frames, "animatelcm.gif")