Update finetune_sample/finetune.py

finetune_sample/finetune.py

@@ -74,7 +74,7 @@ class ScriptArguments:
     lora_r: Optional[int] = field(default=64)
     max_seq_length: Optional[int] = field(default=512)
     model_name: Optional[str] = field(
-        default="dahara1/weblab-10b-instruction-sft-GPTQ/finetune_sample",
+        default="./",
         metadata={
             "help": "The model that you want to train from the Hugging Face hub. E.g. gpt2, gpt2-xl, bert, etc."
         }
@@ -154,6 +154,7 @@ def create_and_prepare_model(args):
         args.model_name,
         device_map=device_map,
         use_safetensors=True,
+        local_files_only=True,
         quantization_config= GPTQConfig(bits=4, disable_exllama=True)
     )