Skip to content

Commit

Permalink
Update cli_fill.py (#396)
Browse files Browse the repository at this point in the history
offload to cpu
  • Loading branch information
yaogang2060 authored Jan 9, 2025
1 parent 4c3e53f commit d06f828
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/flux/cli_fill.py
Original file line number Diff line number Diff line change
Expand Up @@ -271,7 +271,7 @@ def main(
)
opts.seed = None
if offload:
t5, clip, ae = t5.to(torch_device), clip.to(torch_device), ae.to(torch.device)
t5, clip, ae = t5.to(torch_device), clip.to(torch_device), ae.to(torch_device)
inp = prepare_fill(
t5,
clip,
Expand Down

0 comments on commit d06f828

Please sign in to comment.