Update app.py

app.py

@@ -144,7 +144,6 @@ def load_and_prepare_model():
     pipe.scheduler = sched
     #pipe.vae.do_resize=False
     #pipe.vae.vae_scale_factor=8
-    pipe.to(device=device, dtype=torch.bfloat16)
     pipe.vae.set_default_attn_processor()
     #pipe.to(device)
     #pipe.to(torch.bfloat16)
@@ -163,6 +162,7 @@ def load_and_prepare_model():
 
     pipe.unet = pipe.unet.to(memory_format=torch.contiguous_format)
     pipe.unet = torch.compile(pipe.unet, backend="hidet")
+    pipe.to(device=device, dtype=torch.bfloat16)
 
     return pipe
     
@@ -184,6 +184,7 @@ hidet.option.cache_dir("local_cache")
 #hidet.torch.dynamo_config.use_fp16_reduction(True)
 # use tensorcore
 hidet.torch.dynamo_config.use_tensor_core()
+#hidet.torch.dynamo_config.steal_weights(False)
 
 pipe = load_and_prepare_model()