fix: edit was broken by recent changes

README.md

@@ -56,7 +56,7 @@ with prompt-based masking.
 # enter imaginairy shell
 >> aimg
 🤖🧠> edit scenic_landscape.jpg -p "make it winter" --prompt-strength 20
-🤖🧠> edit scenic_landscape.jpg -p "make it winter" --steps 30 --arg-schedule "prompt_strength[2:25:0.5]" --compilation-anim
+🤖🧠> edit scenic_landscape.jpg -p "make it winter" --steps 30 --arg-schedule "prompt_strength[2:25:0.5]" --compilation-anim gif
 🤖🧠> edit dog.jpg -p "make the dog red" --prompt-strength 5
 🤖🧠> edit bowl_of_fruit.jpg -p "replace the fruit with strawberries"
 🤖🧠> edit freckled_woman.jpg -p "make her a cyborg" --prompt-strength 13

imaginairy/api.py

@@ -408,7 +408,7 @@ def _generate_single_image(
 
             elif model.cond_stage_key == "edit":
                 # pix2pix model
-                c_cat = [model.encode_first_stage(init_image_t).mode()]
+                c_cat = [model.encode_first_stage(init_image_t)]
                 c_cat_neutral = [torch.zeros_like(init_latent)]
                 denoiser_cls = CFGEditingDenoiser
             if c_cat:

imaginairy/cmds.py

@@ -533,6 +533,7 @@ def edit_image(  # noqa
     Same as calling `aimg imagine --model edit --init-image my-dog.jpg --init-image-strength 1` except this command
     can batch edit images.
     """
+    allow_compose_phase = False
     return _imagine_cmd(
         ctx,
         prompt,
@@ -557,6 +558,7 @@ def edit_image(  # noqa
         tile,
         tile_x,
         tile_y,
+        allow_compose_phase,
         mask_image,
         mask_prompt,
         mask_mode,