TextToVideo-Dalle

Running

Ruslan Magana Vsevolodovna commited on Aug 21, 2022

Commit

d015fcd

1 Parent(s): 1708b8b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -66,8 +66,8 @@ def get_output_video(text):
           is_verbose=True,
           #dtype=torch.float16 if fp16 else torch.float32
           dtype=torch.float16,
-          device='cuda'
-      )
       image = model.generate_image(
           text,

           is_verbose=True,
           #dtype=torch.float16 if fp16 else torch.float32
           dtype=torch.float16,
+          device='cpu' #'cuda'
+      )
       image = model.generate_image(
           text,