Update app_more1.py

app_more1.py

@@ -42,7 +42,7 @@ pipe = NAGWanPipeline.from_pretrained(
     MODEL_ID, vae=vae, torch_dtype=torch.bfloat16
 )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=5.0)
-pipe.to("cuda")
+
 
 causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
 pipe.load_lora_weights(causvid_path, adapter_name="causvid_lora")
@@ -52,6 +52,7 @@ for name, param in pipe.transformer.named_parameters():
         if "blocks.0" in name:
             param.data = param.data * 0.25
 pipe.fuse_lora()
+pipe.to("cuda")
 pipe.unload_lora_weights()
 
 pipe.transformer.__class__.attn_processors = NagWanTransformer3DModel.attn_processors