Update app.py

app.py

@@ -1,11 +1,18 @@
+from transformers import AutoModelForMaskedLM, AutoTokenizer
+from huggingface_hub import notebook_login
+notebook_login()
+checkpoint = "ziqingyang/chinese-alpaca-2-7b"
+tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+model = AutoModelForMaskedLM.from_pretrained(checkpoint)
+
 !CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \
     --stage sft \
-    --model_name_or_path THUDM/chatglm2-6b \ #ziqingyang/chinese-alpaca-2-7b
+    --model_name_or_path model \
     --do_train \
     --dataset mydata \
     --template chatglm2 \
     --finetuning_type lora \
-    --lora_target query_key_value \
+    --lora_target q_proj,v_proj \
     --output_dir ckpoint \
     --overwrite_cache \
     --auto_find_batch_size=False \
@@ -18,3 +25,7 @@
     --num_train_epochs 3.0 \
     --plot_loss \
     --fp16
+
+
+model.push_to_hub("gutalk_classification") 
+tokenizer.push_to_hub("gutalk_classification")