Skip to content

Commit

Permalink
update wan example
Browse files Browse the repository at this point in the history
  • Loading branch information
richagadgil committed Mar 6, 2025
1 parent 1144c6c commit 6119f43
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions examples/dynamo/wan/wan.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,11 +24,8 @@ def run(args):

pipe = pipe.to("cuda")

print("compile encoder")
pipe.text_encoder = torch.compile(pipe.text_encoder, backend='migraphx')
print("compile transaformer")
pipe.transformer = torch.compile(pipe.transformer, backend='migraphx')
print("compile vae")
pipe.vae.decoder = torch.compile(pipe.vae.decoder, backend='migraphx')

prompt = "A cat walks on the grass, realistic"
Expand Down

0 comments on commit 6119f43

Please sign in to comment.