Skip to content

[Unity][Relax]support torch.arange()+ (int) in torch frontend#15917

Merged
Hzfengsy merged 3 commits intoapache:unityfrom
guoyaol:final_fix_arange
Oct 14, 2023
Merged

[Unity][Relax]support torch.arange()+ (int) in torch frontend#15917
Hzfengsy merged 3 commits intoapache:unityfrom
guoyaol:final_fix_arange

Conversation

@guoyaol
Copy link
Copy Markdown
Contributor

@guoyaol guoyaol commented Oct 11, 2023

In Stable Diffusion XL,
there are some structures, like
a = torch.arange(b) +1

by output the emmited relax.Var, we can support them

@guoyaol guoyaol changed the title support torch.arange()+ (int) in dynamo [Unity][Relax]support torch.arange()+ (int) in torch frontend Oct 11, 2023
@Hzfengsy Hzfengsy merged commit 9892675 into apache:unity Oct 14, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants