Jin Zhu commited on
Commit
3f3b680
·
1 Parent(s): 30f85d9
Files changed (2) hide show
  1. .gitignore +3 -1
  2. src/app.py +2 -4
.gitignore CHANGED
@@ -1 +1,3 @@
1
- cache/*
 
 
 
1
+ cache/*
2
+ src/__pycache__/*
3
+ src/feedback_data/*
src/app.py CHANGED
@@ -117,19 +117,17 @@ FEEDBACK_DATASET_ID = os.environ.get('FEEDBACK_DATASET_ID', 'mamba413/user-feedb
117
  feedback_manager = FeedbackManager(
118
  dataset_repo_id=FEEDBACK_DATASET_ID,
119
  hf_token=os.environ.get('HF_TOKEN'),
120
- local_backup=True # 保留本地备份
121
  )
122
 
123
  # -----------------
124
  # Configuration
125
  # -----------------
126
  MODEL_CONFIG = {
127
- # 'from_pretrained': 'mamba413/AdaDetectGPT-Model' if os.environ.get('SPACE_ID') else './src/FineTune/ckpt/lr_0.0001_a_1',
128
  'from_pretrained': './src/FineTune/ckpt/',
129
  'base_model': 'gemma-1b',
130
  'cache_dir': '../cache',
131
- # 'device': 'mps',
132
- 'device': 'cpu',
133
  # 'device': 'cuda',
134
  }
135
 
 
117
  feedback_manager = FeedbackManager(
118
  dataset_repo_id=FEEDBACK_DATASET_ID,
119
  hf_token=os.environ.get('HF_TOKEN'),
120
+ local_backup=False if os.environ.get('SPACE_ID') else True # 保留本地备份
121
  )
122
 
123
  # -----------------
124
  # Configuration
125
  # -----------------
126
  MODEL_CONFIG = {
 
127
  'from_pretrained': './src/FineTune/ckpt/',
128
  'base_model': 'gemma-1b',
129
  'cache_dir': '../cache',
130
+ 'device': 'cpu' if os.environ.get('SPACE_ID') else 'mps',
 
131
  # 'device': 'cuda',
132
  }
133