Update my_model/fine_tuner/fine_tuner.py
Browse files
my_model/fine_tuner/fine_tuner.py
CHANGED
@@ -51,7 +51,7 @@ class QLoraConfig:
|
|
51 |
Initializes QLoraConfig with specific LoRA parameters.
|
52 |
|
53 |
"""
|
54 |
-
# please refer to config file 'fine_tuning_config.py' for QLORA arguments description.
|
55 |
self.lora_config = LoraConfig(
|
56 |
lora_alpha=config.LORA_ALPHA,
|
57 |
lora_dropout=config.LORA_DROPOUT,
|
|
|
51 |
Initializes QLoraConfig with specific LoRA parameters.
|
52 |
|
53 |
"""
|
54 |
+
# please refer to config file 'my_model/config/fine_tuning_config.py' for QLORA arguments description.
|
55 |
self.lora_config = LoraConfig(
|
56 |
lora_alpha=config.LORA_ALPHA,
|
57 |
lora_dropout=config.LORA_DROPOUT,
|