Spaces:
Sleeping
Sleeping
Upload run_cloud_training.py with huggingface_hub
Browse files- run_cloud_training.py +1 -1
run_cloud_training.py
CHANGED
@@ -539,7 +539,7 @@ def train(config_path, dataset_name, output_dir, upload_to_hub=False, hub_repo_n
|
|
539 |
logger.info("Using Unsloth-specific LoRA target modules")
|
540 |
|
541 |
lora_config_obj = LoraConfig(
|
542 |
-
r=lora_config.get("r",
|
543 |
lora_alpha=lora_config.get("lora_alpha", 32),
|
544 |
lora_dropout=lora_config.get("lora_dropout", 0.05),
|
545 |
bias=lora_config.get("bias", "none"),
|
|
|
539 |
logger.info("Using Unsloth-specific LoRA target modules")
|
540 |
|
541 |
lora_config_obj = LoraConfig(
|
542 |
+
r=lora_config.get("r", 8),
|
543 |
lora_alpha=lora_config.get("lora_alpha", 32),
|
544 |
lora_dropout=lora_config.get("lora_dropout", 0.05),
|
545 |
bias=lora_config.get("bias", "none"),
|