Update app.py

app.py

@@ -223,11 +223,12 @@ def start_training(
 
     # Update the config with user inputs
     config["config"]["name"] = slugged_lora_name
-    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
+    #if(is_spaces):
+    #    config["config"]["process"][0]["model"]["low_vram"] = False
+    #    config["config"]["process"][0]["train"]["batch_size"] = 2
+    #    config["config"]["process"][0]["train"]["steps"] = int(steps)/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)