Steveeeeeeen HF Staff commited on
Commit
b6a58c5
·
verified ·
1 Parent(s): 717bbe8

Update stepaudio2.py

Browse files
Files changed (1) hide show
  1. stepaudio2.py +1 -1
stepaudio2.py CHANGED
@@ -1,6 +1,6 @@
1
  import torch
2
  from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
3
-
4
  from utils import compute_token_num, load_audio, log_mel_spectrogram, padding_mels
5
 
6
  hf_token = os.getenv("HF_TOKEN", None)
 
1
  import torch
2
  from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
3
+ import os
4
  from utils import compute_token_num, load_audio, log_mel_spectrogram, padding_mels
5
 
6
  hf_token = os.getenv("HF_TOKEN", None)