Spaces:

mazpie
/

genrl

Running on Zero

mazpie commited on Jun 27

Commit

4943924

•

1 Parent(s): ff88751

Update demo/t2v.py

Files changed (1) hide show

demo/t2v.py CHANGED Viewed

@@ -96,7 +96,6 @@ class Text2Video():
         # Offload GPU
         self.agent.to('cpu')
         self.clip.to('cpu')
-        self.tokenizer.to('cpu')
         return os.path.join(self.result_dir, f"{prompt_str}.mp4")
     def download_model(self, model_folder, model_filename):

         # Offload GPU
         self.agent.to('cpu')
         self.clip.to('cpu')
         return os.path.join(self.result_dir, f"{prompt_str}.mp4")
     def download_model(self, model_folder, model_filename):