rahul7star commited on
Commit
400b637
·
verified ·
1 Parent(s): 0877c51

Update app_flash.py

Browse files
Files changed (1) hide show
  1. 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 = 1000,
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...")