update

hf_demo.py

@@ -51,7 +51,7 @@ def demo_inference_gen_artistic(adapter_choice:str, prompt:str, seed:int=0, step
                             height=512, width=512, scales=[adapter_scale],
                             save_dir=None, seed=seed,steps=steps, guidance_scale=guidance_scale,
                             start_noise=-1, show=False, style_prompt=style_prompt, no_load=True,
-                            from_scratch=True, device=device, weight_dtype=dtype)[0][1.0][0]
+                            from_scratch=True, device=device, weight_dtype=dtype)[0][adapter_scale][0]
     return pred_images
 
 @spaces.GPU
@@ -101,7 +101,7 @@ def demo_inference_stylization_artistic(ref_image, adapter_choice:str, prompt:st
                             height=512, width=512, scales=[adapter_scale],
                             save_dir=None, seed=seed,steps=steps, guidance_scale=guidance_scale,
                             start_noise=start_noise, show=False, style_prompt=style_prompt, no_load=True,
-                            from_scratch=False, device=device, weight_dtype=dtype)[0][1.0][0]
+                            from_scratch=False, device=device, weight_dtype=dtype)[0][adapter_scale][0]
     return pred_images