Update app.py

app.py

@@ -213,13 +213,19 @@ trainer = QuestionAnsweringTrainer(
 
 #################################################
 # special Trainer Baize Model
-# Parameters
-MICRO_BATCH_SIZE = int(sys.argv[2])
+# Parameters für Model 7b: 7b 32 0.0002
+# Model 13b: 13b 16 0.0001
+# Model 30b: 30b 8 0.00005
+arg1 = '7b'
+arg2 = 32
+arg3 = 0.0002
+
+MICRO_BATCH_SIZE = int(arg2)
 BATCH_SIZE = 64
-size = sys.argv[1]
+size = arg1
 GRADIENT_ACCUMULATION_STEPS = BATCH_SIZE // MICRO_BATCH_SIZE
 EPOCHS = 1
-LEARNING_RATE = float(sys.argv[3])
+LEARNING_RATE = float(arg3)
 CUTOFF_LEN = 512
 LORA_R = 8
 LORA_ALPHA = 16