Update orpheus-tts/engine_class.py

orpheus-tts/engine_class.py

@@ -5,7 +5,7 @@ from vllm import AsyncLLMEngine, AsyncEngineArgs, SamplingParams
 from transformers import AutoTokenizer
 import threading
 import queue
-from .decoder import tokens_decoder_sync
+from decoder import tokens_decoder_sync
 
 class OrpheusModel:
     def __init__(self, model_name, dtype=torch.bfloat16, tokenizer=None, **engine_kwargs):
@@ -63,7 +63,7 @@ class OrpheusModel:
         if (model_name  in unsupported_models):
             raise ValueError(f"Model {model_name} is not supported. Only medium-3b is supported, small, micro and nano models will be released very soon")
         elif model_name in model_map:
-            return model_name[model_name]["repo_id"]
+            return model_map[model_name]["repo_id"]
         else:
             return model_name