up
This commit is contained in:
parent
5d40de00fb
commit
676e6acfef
1 changed files with 2 additions and 2 deletions
|
@ -9,7 +9,7 @@ model:
|
||||||
timesteps: 1000
|
timesteps: 1000
|
||||||
first_stage_key: "jpg"
|
first_stage_key: "jpg"
|
||||||
cond_stage_key: "txt"
|
cond_stage_key: "txt"
|
||||||
image_size: 32
|
image_size: 64
|
||||||
channels: 4
|
channels: 4
|
||||||
cond_stage_trainable: false # Note: different from the one we trained before
|
cond_stage_trainable: false # Note: different from the one we trained before
|
||||||
conditioning_key: crossattn
|
conditioning_key: crossattn
|
||||||
|
@ -28,7 +28,7 @@ model:
|
||||||
unet_config:
|
unet_config:
|
||||||
target: ldm.modules.diffusionmodules.openaimodel.UNetModel
|
target: ldm.modules.diffusionmodules.openaimodel.UNetModel
|
||||||
params:
|
params:
|
||||||
image_size: 32
|
image_size: 32 # unused
|
||||||
in_channels: 4
|
in_channels: 4
|
||||||
out_channels: 4
|
out_channels: 4
|
||||||
model_channels: 320
|
model_channels: 320
|
||||||
|
|
Loading…
Reference in a new issue