Update demo/t2v.py

demo/t2v.py

@@ -57,6 +57,8 @@ class Text2Video():
         torch.cuda.empty_cache()
         self.agent.to('cuda')
         self.clip.to('cuda')
+        self.tokenizer.to('cuda')
+
         print('start:', prompt, time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time())))
         start = time.time()
 
@@ -95,6 +97,7 @@ 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):