improve configs for better gpu

app.py

@@ -223,7 +223,11 @@ def start_training(
 
     # Update the config with user inputs
     config["config"]["name"] = slugged_lora_name
-    config["config"]["process"][0]["model"]["low_vram"] = True
+    if(is_spaces):
+        config["config"]["process"][0]["model"]["low_vram"] = False
+        config["config"]["process"][0]["train"]["batch_size"] = 2
+    else:
+        config["config"]["process"][0]["model"]["low_vram"] = True
     config["config"]["process"][0]["train"]["skip_first_sample"] = True
     config["config"]["process"][0]["train"]["steps"] = int(steps)
     config["config"]["process"][0]["train"]["lr"] = float(lr)