lysandre HF staff commited on
Commit
deef182
1 Parent(s): 5018b93

Update text_to_video.py

Browse files
Files changed (1) hide show
  1. text_to_video.py +2 -1
text_to_video.py CHANGED
@@ -1,9 +1,10 @@
1
  import torch
2
  from typing import Any
3
- from transformers.tools.base import Tool
4
  from transformers.utils import is_accelerate_available
5
 
6
  from diffusers import DiffusionPipeline
 
7
  if is_accelerate_available():
8
  from accelerate import PartialState
9
 
 
1
  import torch
2
  from typing import Any
3
+ from transformers.agents.tools import Tool
4
  from transformers.utils import is_accelerate_available
5
 
6
  from diffusers import DiffusionPipeline
7
+
8
  if is_accelerate_available():
9
  from accelerate import PartialState
10