ford442 commited on
Commit
062249f
·
verified ·
1 Parent(s): 4078146

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -13,8 +13,10 @@ login(token=os.environ.get("HF_TOKEN"))
13
  def install_torch280cu129():
14
  subprocess.run(['sh', './torch.sh'])
15
  #install_torch280cu129()
 
16
  token=os.environ.get("HF_TOKEN")
17
- @spaces.GPU(duration=120)
 
18
  def install_flashattn():
19
  subprocess.run(['sh', './flashattn.sh'])
20
  install_flashattn()
 
13
  def install_torch280cu129():
14
  subprocess.run(['sh', './torch.sh'])
15
  #install_torch280cu129()
16
+
17
  token=os.environ.get("HF_TOKEN")
18
+
19
+ @spaces.GPU(required=True)
20
  def install_flashattn():
21
  subprocess.run(['sh', './flashattn.sh'])
22
  install_flashattn()