Spaces:
Sleeping
Sleeping
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,
|