tcy6 commited on
Commit
357ea73
1 Parent(s): 415a781

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -18,6 +18,7 @@ import json
18
 
19
  cache_dir = '/data/KB'
20
  os.makedirs(cache_dir, exist_ok=True)
 
21
 
22
  def weighted_mean_pooling(hidden, attention_mask):
23
  attention_mask_ = attention_mask * attention_mask.cumsum(dim=1)
 
18
 
19
  cache_dir = '/data/KB'
20
  os.makedirs(cache_dir, exist_ok=True)
21
+ os.system(f"rm -rf /data-nvme")
22
 
23
  def weighted_mean_pooling(hidden, attention_mask):
24
  attention_mask_ = attention_mask * attention_mask.cumsum(dim=1)