Update config.py
Browse files
config.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
from transformers import PretrainedConfig
|
2 |
|
3 |
class SpeechLLMModelConfig(PretrainedConfig):
|
@@ -11,4 +12,4 @@ class SpeechLLMModelConfig(PretrainedConfig):
|
|
11 |
self.audio_processor_name = "facebook/hubert-large-ls960-ft"
|
12 |
self.audio_encoder_name = 'facebook/hubert-xlarge-ll60k'
|
13 |
self.llm_model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
14 |
-
self.llm_model_checkpoint = "
|
|
|
1 |
+
import os
|
2 |
from transformers import PretrainedConfig
|
3 |
|
4 |
class SpeechLLMModelConfig(PretrainedConfig):
|
|
|
12 |
self.audio_processor_name = "facebook/hubert-large-ls960-ft"
|
13 |
self.audio_encoder_name = 'facebook/hubert-xlarge-ll60k'
|
14 |
self.llm_model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
15 |
+
self.llm_model_checkpoint = os.path.join(os.getcwd(), "lora_checkpoint")
|