Jin Zhu commited on
Commit
ecfa074
·
1 Parent(s): a9c7cad

add checkpoint file

Browse files
Files changed (1) hide show
  1. src/app.py +1 -1
src/app.py CHANGED
@@ -85,7 +85,7 @@ def load_model(from_pretrained, base_model, cache_dir, device):
85
  # Configuration
86
  # -----------------
87
  MODEL_CONFIG = {
88
- 'from_pretrained': str(APP_DIR / 'FineTune' / 'ckpt' / 'lr_0.0001_a_1'),
89
  'base_model': 'gemma-1b',
90
  'cache_dir': '../cache',
91
  'device': 'mps',
 
85
  # Configuration
86
  # -----------------
87
  MODEL_CONFIG = {
88
+ 'from_pretrained': 'mamba413/AdaDetectGPT-Model',
89
  'base_model': 'gemma-1b',
90
  'cache_dir': '../cache',
91
  'device': 'mps',