jw2yang commited on
Commit
36894e0
·
1 Parent(s): 0ac095e
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ from PIL import Image
7
  from transformers import AutoModelForCausalLM, AutoProcessor
8
  import re
9
  import random
 
10
 
11
  # add a command for installing flash-attn
12
  os.system('pip install flash-attn --no-build-isolation')
 
7
  from transformers import AutoModelForCausalLM, AutoProcessor
8
  import re
9
  import random
10
+ import os
11
 
12
  # add a command for installing flash-attn
13
  os.system('pip install flash-attn --no-build-isolation')