firstkillday commited on
Commit
d5d6ffc
·
verified ·
1 Parent(s): 261218f

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -51,11 +51,9 @@ from PIL import Image
51
  from moviepy.editor import VideoFileClip, AudioFileClip
52
  from pydub import AudioSegment
53
  from huggingface_hub import snapshot_download
54
- try:
55
- from torchao.quantization import quantize_, int8_weight_only
56
- except ImportError:
57
- quantize_ = None
58
- int8_weight_only = None
59
 
60
  from src.models.unet_2d_condition import UNet2DConditionModel
61
  from src.models.unet_3d_emo import EMOUNet3DConditionModel
 
51
  from moviepy.editor import VideoFileClip, AudioFileClip
52
  from pydub import AudioSegment
53
  from huggingface_hub import snapshot_download
54
+ # torchao removed for CPU mode
55
+ quantize_ = None
56
+ int8_weight_only = None
 
 
57
 
58
  from src.models.unet_2d_condition import UNet2DConditionModel
59
  from src.models.unet_3d_emo import EMOUNet3DConditionModel