Spaces:
Running
Running
Jin Zhu commited on
Commit ·
3f3b680
1
Parent(s): 30f85d9
update
Browse files- .gitignore +3 -1
- 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 |
-
|
| 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 |
|