apply lint
This commit is contained in:
parent
d3e8caade2
commit
841b18576c
|
@ -126,9 +126,7 @@ def convert_diffusion_stable(
|
||||||
# UNET
|
# UNET
|
||||||
if single_vae:
|
if single_vae:
|
||||||
unet_inputs = ["sample", "timestep", "encoder_hidden_states", "class_labels"]
|
unet_inputs = ["sample", "timestep", "encoder_hidden_states", "class_labels"]
|
||||||
unet_scale = torch.tensor(4).to(
|
unet_scale = torch.tensor(4).to(device=ctx.training_device, dtype=torch.long)
|
||||||
device=ctx.training_device, dtype=torch.long
|
|
||||||
)
|
|
||||||
else:
|
else:
|
||||||
unet_inputs = ["sample", "timestep", "encoder_hidden_states", "return_dict"]
|
unet_inputs = ["sample", "timestep", "encoder_hidden_states", "return_dict"]
|
||||||
unet_scale = torch.tensor(False).to(
|
unet_scale = torch.tensor(False).to(
|
||||||
|
|
Loading…
Reference in New Issue