mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-08 22:48:25 -05:00
hotfix: sdxl corealize
This commit is contained in:
@@ -356,8 +356,7 @@ class DPMPP2MSampler:
|
||||
c=c,
|
||||
uc=uc,
|
||||
)
|
||||
x.realize()
|
||||
old_denoised.realize()
|
||||
x.realize(old_denoised)
|
||||
|
||||
return x
|
||||
|
||||
@@ -396,8 +395,7 @@ if __name__ == "__main__":
|
||||
|
||||
c, uc = model.create_conditioning([args.prompt], args.width, args.height)
|
||||
del model.conditioner
|
||||
for v in c .values(): v.realize()
|
||||
for v in uc.values(): v.realize()
|
||||
Tensor.realize(*c.values(), *uc.values())
|
||||
print("created batch")
|
||||
|
||||
# https://github.com/Stability-AI/generative-models/blob/fbdc58cab9f4ee2be7a5e1f2e2787ecd9311942f/sgm/inference/helpers.py#L101
|
||||
|
||||
Reference in New Issue
Block a user