Update app.py

app.py

@@ -83,7 +83,7 @@ with torch.no_grad():  # enable image inputs
         new_img_in.proj.bias.copy_(pipe.transformer.x_embedder.proj.bias)
     pipe.transformer.x_embedder = new_img_in
 
-lora_state_dict = safetensors.torch.load_file(lora_path, device="cpu")
+lora_state_dict = safetensors.torch.load_file(lora_path)
 transformer_lora_state_dict = {f'{k.replace("transformer.", "")}': v for k, v in lora_state_dict.items() if k.startswith("transformer.") and "lora" in k}
 
 pipe.load_lora_into_transformer(transformer_lora_state_dict, transformer=pipe.transformer, adapter_name="i2v", _pipeline=pipe)