alexkueck commited on
Commit
456a3d3
·
1 Parent(s): 764dfa4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -27,8 +27,10 @@ from peft import (
27
  ####################################################
28
  # Konstanten
29
  ####################################################
30
- #Konstanten speziell für Baize Model:
31
  OUTPUT_DIR = "alexkueck/li-tis-tuned-2"
 
 
 
32
  MICRO_BATCH_SIZE = int(arg2)
33
  BATCH_SIZE = 64
34
  size = arg1
@@ -48,7 +50,7 @@ TARGET_MODULES = [
48
  "gate_proj",
49
  "up_proj",
50
  ]
51
-
52
  #####################################################
53
  #Hilfsfunktionen für das Training
54
  #####################################################
 
27
  ####################################################
28
  # Konstanten
29
  ####################################################
 
30
  OUTPUT_DIR = "alexkueck/li-tis-tuned-2"
31
+
32
+ '''
33
+ #Konstanten speziell für Baize Model:
34
  MICRO_BATCH_SIZE = int(arg2)
35
  BATCH_SIZE = 64
36
  size = arg1
 
50
  "gate_proj",
51
  "up_proj",
52
  ]
53
+ '''
54
  #####################################################
55
  #Hilfsfunktionen für das Training
56
  #####################################################