Spaces:
Paused
Paused
Update app.py
Browse files
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 |
#####################################################
|