hanAlex commited on
Commit
15c76a9
β€’
1 Parent(s): 41155a0

Update web_demo.py

Browse files
Files changed (1) hide show
  1. web_demo.py +1 -1
web_demo.py CHANGED
@@ -42,7 +42,7 @@ if __name__ == "__main__":
42
  flow_checkpoint = os.path.join(args.flow_path, 'flow.pt')
43
  hift_checkpoint = os.path.join(args.flow_path, 'hift.pt')
44
  glm_tokenizer = None
45
- device = "cuda"
46
  audio_decoder: AudioDecoder = None
47
  whisper_model, feature_extractor = None, None
48
 
 
42
  flow_checkpoint = os.path.join(args.flow_path, 'flow.pt')
43
  hift_checkpoint = os.path.join(args.flow_path, 'hift.pt')
44
  glm_tokenizer = None
45
+ device = "cpu"
46
  audio_decoder: AudioDecoder = None
47
  whisper_model, feature_extractor = None, None
48