Spaces:
Running
on
Zero
Running
on
Zero
File size: 2,470 Bytes
9e426da |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import torch
def simple_guidance_fn(out, cfg):
uncondition, condtion = out.chunk(2, dim=0)
out = uncondition + cfg * (condtion - uncondition)
return out
def c3_guidance_fn(out, cfg):
# guidance function in DiT/SiT, seems like a bug not a feature?
uncondition, condtion = out.chunk(2, dim=0)
out = condtion
out[:, :3] = uncondition[:, :3] + cfg * (condtion[:, :3] - uncondition[:, :3])
return out
def c4_guidance_fn(out, cfg):
# guidance function in DiT/SiT, seems like a bug not a feature?
uncondition, condition = out.chunk(2, dim=0)
out = condition
out[:, :4] = uncondition[:, :4] + cfg * (condition[:, :4] - uncondition[:, :4])
out[:, 4:] = uncondition[:, 4:] + 1.05 * (condition[:, 4:] - uncondition[:, 4:])
return out
def c4_p05_guidance_fn(out, cfg):
# guidance function in DiT/SiT, seems like a bug not a feature?
uncondition, condition = out.chunk(2, dim=0)
out = condition
out[:, :4] = uncondition[:, :4] + cfg * (condition[:, :4] - uncondition[:, :4])
out[:, 4:] = uncondition[:, 4:] + 1.05 * (condition[:, 4:] - uncondition[:, 4:])
return out
def c4_p10_guidance_fn(out, cfg):
# guidance function in DiT/SiT, seems like a bug not a feature?
uncondition, condition = out.chunk(2, dim=0)
out = condition
out[:, :4] = uncondition[:, :4] + cfg * (condition[:, :4] - uncondition[:, :4])
out[:, 4:] = uncondition[:, 4:] + 1.10 * (condition[:, 4:] - uncondition[:, 4:])
return out
def c4_p15_guidance_fn(out, cfg):
# guidance function in DiT/SiT, seems like a bug not a feature?
uncondition, condition = out.chunk(2, dim=0)
out = condition
out[:, :4] = uncondition[:, :4] + cfg * (condition[:, :4] - uncondition[:, :4])
out[:, 4:] = uncondition[:, 4:] + 1.15 * (condition[:, 4:] - uncondition[:, 4:])
return out
def c4_p20_guidance_fn(out, cfg):
# guidance function in DiT/SiT, seems like a bug not a feature?
uncondition, condition = out.chunk(2, dim=0)
out = condition
out[:, :4] = uncondition[:, :4] + cfg * (condition[:, :4] - uncondition[:, :4])
out[:, 4:] = uncondition[:, 4:] + 1.20 * (condition[:, 4:] - uncondition[:, 4:])
return out
def p4_guidance_fn(out, cfg):
# guidance function in DiT/SiT, seems like a bug not a feature?
uncondition, condtion = out.chunk(2, dim=0)
out = condtion
out[:, 4:] = uncondition[:, 4:] + cfg * (condtion[:, 4:] - uncondition[:, 4:])
return out
|