Update app_flash.py
Browse files- app_flash.py +1 -1
app_flash.py
CHANGED
|
@@ -90,7 +90,7 @@ def push_flashpack_model_to_hf(model, hf_repo: str):
|
|
| 90 |
# ============================================================
|
| 91 |
def train_flashpack_model(
|
| 92 |
dataset_name: str = "gokaygokay/prompt-enhancer-dataset",
|
| 93 |
-
max_encode: int =
|
| 94 |
device: str = "cpu"
|
| 95 |
) -> Tuple[GemmaTrainer, object, object, object, torch.Tensor]:
|
| 96 |
print("📦 Loading dataset...")
|
|
|
|
| 90 |
# ============================================================
|
| 91 |
def train_flashpack_model(
|
| 92 |
dataset_name: str = "gokaygokay/prompt-enhancer-dataset",
|
| 93 |
+
max_encode: int = 5000,
|
| 94 |
device: str = "cpu"
|
| 95 |
) -> Tuple[GemmaTrainer, object, object, object, torch.Tensor]:
|
| 96 |
print("📦 Loading dataset...")
|