fixing training

app.py

@@ -233,10 +233,8 @@ class Demo:
 #             positive_prompt = ''
 #         if negative_prompt is None:
 #             negative_prompt = ''
-#         if is_person is None:
-#             is_person = False
-#         else:
-#             is_person = True
+          if attributes_input == '':
+              attributes_input = None
         print(target_concept, positive_prompt, negative_prompt, attributes_input, is_person)
         
         randn = torch.randint(1, 10000000, (1,)).item()

trainscripts/textsliders/demotrain.py

@@ -432,5 +432,5 @@ def train_xl(target, positive, negative, lr, iterations, config_file, rank, devi
     
     prompts = prompt_util.load_prompts_from_yaml(path=config.prompts_file, target=target, positive=positive, negative=negative,  attributes=attributes)
     
-    device = torch.device(f"cuda:{device}")
+    device = torch.device(device)
     train(config, prompts, device)