Picsart-AI-Research / Text2Video-Zero

[ICCV 2023 Oral] Text-to-Image Diffusion Models are Zero-Shot Video Generators
https://text2video-zero.github.io/
Other
3.91k stars 336 forks source link

AssertionError: Torch not compiled with CUDA enabled #76

Open xspace-solution opened 4 weeks ago

xspace-solution commented 4 weeks ago

I wrote simple code to generate video.

import torch
from model import Model

model = Model(device = "cuda", dtype = torch.float16)

prompt = "A horse galloping on a street"
params = {"t0": 44, "t1": 47 , "motion_field_strength_x" : 12, "motion_field_strength_y" : 12, "video_length": 8}

out_path, fps = f"./text2video_{prompt.replace(' ','_')}.mp4", 4
model.process_text2video(prompt, fps = fps, path = out_path, **params)

But I am getting this issue after I run the script. AssertionError: Torch not compiled with CUDA enabled

Tabz05 commented 2 weeks ago

Hi!

Did you set up the project according to the instructions provided?