fix numpy import error after tensor2vid refactor in huggingface#6626

a-r-r-o-w · Jan 23, 2024 · 032c24f · 032c24f
1 parent 5674a71
commit 032c24f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/diffusers/pipelines/animatediff/pipeline_animatediff.py b/src/diffusers/pipelines/animatediff/pipeline_animatediff.py
@@ -16,6 +16,7 @@
 import math
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
+import numpy as np
 import torch
 import torch.fft as fft
 from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer, CLIPVisionModelWithProjection