Update generate.yaml

update
2024-08-14 11:50:29 +08:00
parent 8bb362a853
commit f4b87afae5
1 changed files with 0 additions and 1 deletions
@@ -9,7 +9,6 @@ model:
    timesteps: 1000
    first_stage_key: edited
    cond_stage_key: edit
-    first_stage_downsample: True
    image_size: 16
    channels: 4
    cond_stage_trainable: false   # Note: different from the one we trained before