Siddhant commited on
Commit
20a87d9
1 Parent(s): 8ed40d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ blocksize = 512
51
  tts_model.tts_to_file("text", speaker_id, quiet=True)
52
  dummy_input = torch.randn(
53
  (1, 80, 3000),
54
- dtype="float16",
55
  device="cuda",
56
  )
57
  transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-tiny.en", device="cuda")
 
51
  tts_model.tts_to_file("text", speaker_id, quiet=True)
52
  dummy_input = torch.randn(
53
  (1, 80, 3000),
54
+ dtype=getattr(torch, "float16"),
55
  device="cuda",
56
  )
57
  transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-tiny.en", device="cuda")