text

app.py

@@ -186,10 +186,8 @@ def infer(
         if num_of_interpolation == 3:
             # Addition or subtraction mode.
             if operation_mode == 'Addition':
-                assert config.prompt_s is None, "Only one of prompt_a or prompt_s should be provided."
                 z_init_temp = _z_init[0] + _z_init[1]
             elif operation_mode == 'Subtraction':
-                assert config.prompt_a is None, "Only one of prompt_a or prompt_s should be provided."
                 z_init_temp = _z_init[0] - _z_init[1]
             else:
                 raise NotImplementedError("Either prompt_a or prompt_s must be provided for 3-sample mode.")