Update src/facerender/animate.py

src/facerender/animate.py

@@ -120,7 +120,6 @@ class AnimateFromCoeff():
 
         return checkpoint['epoch']
         
-from torch.cuda.amp import autocast
     def generate(self, x, video_save_dir, pic_path, crop_info, enhancer=None, background_enhancer=None, preprocess='crop'):
         source_image=x['source_image'].type(torch.FloatTensor)
         source_semantics=x['source_semantics'].type(torch.FloatTensor)