Upload train_viutranslate.py with huggingface_hub

train_viutranslate.py

@@ -62,8 +62,8 @@ def auto_profile_hardware():
     dtype = torch.bfloat16 if bf16_supported else torch.float16
 
     if vram_gb >= 30:
-        micro_batch, grad_accum = 16, 4
-        desc = "High-Tier Workstation / Datacenter (32GB+)"
+        micro_batch, grad_accum = 32, 2
+        desc = "NVIDIA RTX 5090 / High-Tier 32GB Blackwell Beast"
     elif vram_gb >= 20:
         micro_batch, grad_accum = 12, 6
         desc = "Pro-Tier GPU (24GB)"