qinghuazhou commited on
Commit
ec5e0f8
·
1 Parent(s): fd0a592

use of cache in persistent storage

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -11,7 +11,7 @@ from util import utils
11
 
12
  ## UTILITY FUNCTIONS ################################################
13
 
14
- @spaces.GPU(duration=500)
15
  def load_editor(model_name='gpt2-xl'):
16
 
17
  # loading hyperparameters
@@ -23,7 +23,8 @@ def load_editor(model_name='gpt2-xl'):
23
  hparams = hparams,
24
  layer = 13,
25
  edit_mode='in-place',
26
- verbose=True
 
27
  )
28
  return editor
29
 
 
11
 
12
  ## UTILITY FUNCTIONS ################################################
13
 
14
+ @spaces.GPU(duration=180)
15
  def load_editor(model_name='gpt2-xl'):
16
 
17
  # loading hyperparameters
 
23
  hparams = hparams,
24
  layer = 13,
25
  edit_mode='in-place',
26
+ cache_path='/data/cache/',
27
+ verbose=True
28
  )
29
  return editor
30