Spaces:
Starting
on
T4
Starting
on
T4
FantasticGNU
commited on
Commit
•
17e2e41
1
Parent(s):
566370b
Update app.py
Browse files
app.py
CHANGED
@@ -20,8 +20,7 @@ args = {
|
|
20 |
'max_tgt_len': 128,
|
21 |
'lora_r': 32,
|
22 |
'lora_alpha': 32,
|
23 |
-
'lora_dropout': 0.1
|
24 |
-
'layers': [7,15,23,31]
|
25 |
}
|
26 |
|
27 |
model = OpenLLAMAPEFTModel(**args)
|
@@ -245,4 +244,4 @@ with gr.Blocks() as demo:
|
|
245 |
], show_progress=True)
|
246 |
|
247 |
|
248 |
-
demo.queue().launch(
|
|
|
20 |
'max_tgt_len': 128,
|
21 |
'lora_r': 32,
|
22 |
'lora_alpha': 32,
|
23 |
+
'lora_dropout': 0.1
|
|
|
24 |
}
|
25 |
|
26 |
model = OpenLLAMAPEFTModel(**args)
|
|
|
244 |
], show_progress=True)
|
245 |
|
246 |
|
247 |
+
demo.queue().launch()
|