pan-yl commited on
Commit
380884d
·
1 Parent(s): fd61736
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -3,9 +3,12 @@
3
  import os
4
  import shlex
5
  import subprocess
6
-
7
  subprocess.run(shlex.split('pip install flash-attn --no-build-isolation'), env=os.environ | {'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"})
8
 
 
 
 
 
9
  import argparse
10
  import base64
11
  import copy
 
3
  import os
4
  import shlex
5
  import subprocess
 
6
  subprocess.run(shlex.split('pip install flash-attn --no-build-isolation'), env=os.environ | {'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"})
7
 
8
+ import sys
9
+ import csv
10
+ csv.field_size_limit(sys.maxsize)
11
+
12
  import argparse
13
  import base64
14
  import copy