update

hf_demo.py

@@ -39,14 +39,17 @@ def demo_inference_gen(adapter_choice:str, prompt:str, samples:int=1,seed:int=0,
     adapter_path = lora_map[adapter_choice]
     if adapter_path not in [None, "None"]:
         adapter_path = f"data/Art_adapters/{adapter_path}/adapter_alpha1.0_rank1_all_up_1000steps.pt"
-
+        style_prompt="sks art"
+    else:
+        style_prompt=None
     prompts = [prompt]*samples
     infer_loader = get_validation_dataloader(prompts,num_workers=0)
     network = get_lora_network(pipe.unet, adapter_path, weight_dtype=dtype)["network"]
+
     pred_images = inference(network, pipe.tokenizer, pipe.text_encoder, pipe.vae, pipe.unet, pipe.scheduler, infer_loader,
                             height=512, width=512, scales=[1.0],
                             save_dir=None, seed=seed,steps=steps, guidance_scale=guidance_scale,
-                            start_noise=-1, show=False, style_prompt="sks art", no_load=True,
+                            start_noise=-1, show=False, style_prompt=style_prompt, no_load=True,
                             from_scratch=True, device=device, weight_dtype=dtype)[0][1.0]
     return pred_images
 @spaces.GPU