Fix relative imports to allow direct execution of inference.py

codec.py

@@ -11,7 +11,7 @@ import soundfile as sf
 from pathlib import Path
 from typing import Optional, Union
 
-from .config import (
+from config import (
     CODEC_MODEL_NAME, CODEC_SAMPLE_RATE,
     CODEC_CODEBOOK_SIZE, CODEC_FRAME_RATE,
 )

inference.py

@@ -10,13 +10,13 @@ import torch
 import argparse
 import time
 from pathlib import Path
-from .config import (
+from config import (
     AUDIO_OFFSET, NUM_AUDIO_TOKENS, END_OF_SPEECH_TOKEN_ID,
     START_OF_SPEECH_TOKEN_ID, CODEC_SAMPLE_RATE, CODEC_FRAME_RATE,
 )
-from .tokenizer import TTSTokenizer
-from .codec import CodecV6
-from .model import load_for_inference
+from tokenizer import TTSTokenizer
+from codec import CodecV6
+from model import load_for_inference
 
 
 @torch.no_grad()

model.py

@@ -27,7 +27,7 @@ import torch.nn.functional as F
 from typing import Optional, Tuple, Dict
 from dataclasses import dataclass
 
-from .config import (
+from config import (
     TOTAL_VOCAB_SIZE, ENCODER_VOCAB_SIZE, DECODER_VOCAB_SIZE,
     ENC_D_MODEL, ENC_N_HEADS, ENC_N_LAYERS, ENC_D_FF,
     DEC_D_MODEL, DEC_N_HEADS, DEC_N_LAYERS, DEC_D_FF,

tokenizer.py

@@ -10,7 +10,7 @@ import re
 import torch
 from typing import Optional
 
-from .config import (
+from config import (
     TEXT_CHARS, TEXT_OFFSET, AUDIO_OFFSET,
     SPECIAL_TOKENS, NUM_SPECIAL_TOKENS, CODEC_CODEBOOK_SIZE,
     TOTAL_VOCAB_SIZE,