train-dreambooth-lora-sdxl

Runtime error

fffiloni commited on Sep 16, 2023

Commit

06684f1

1 Parent(s): 2d152a1

ensure gpu is turned off if errors occurs

Files changed (1) hide show

app.py CHANGED Viewed

@@ -143,7 +143,8 @@ def train_dreambooth_lora_sdxl(dataset_id, instance_data_dir, lora_trained_xl_fo
         {str(e)}
         ```
         '''
-        #swap_hardware(hf_token, "cpu-basic")
         #write_to_community(title,description,hf_token)
 def main(dataset_id,

         {str(e)}
         ```
         '''
+        if remove_gpu:
+            swap_hardware(hf_token, "cpu-basic")
         #write_to_community(title,description,hf_token)
 def main(dataset_id,