update

hf_demo.py

@@ -42,7 +42,7 @@ def demo_inference_gen(adapter_choice:str, prompt:str, samples:int=1,seed:int=0,
 
     prompts = [prompt]*samples
     infer_loader = get_validation_dataloader(prompts,num_workers=0)
-    network = get_lora_network(pipe.unet, adapter_path)["network"]
+    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,
@@ -52,7 +52,7 @@ def demo_inference_gen(adapter_choice:str, prompt:str, samples:int=1,seed:int=0,
 @spaces.GPU
 def demo_inference_stylization(adapter_path:str, prompts:list, image:list, start_noise=800,seed:int=0):
     infer_loader = get_validation_dataloader(prompts, image,num_workers=0)
-    network = get_lora_network(pipe.unet, adapter_path,"all_up")["network"]
+    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=[0.,1.],
                             save_dir=None, seed=seed,steps=20, guidance_scale=7.5,