AttributeError: module 'keras.backend' has no attribute 'is_tensor'

[nickwong64]

When running on Colab Pro. Below error occurs: AttributeError: module 'keras.backend' has no attribute 'is_tensor' Any idea how to fix?

Run training

!(python main.py \ -t \ --base configs/stable-diffusion/pokemon.yaml \ --gpus "$gpu_list" \ --scale_lr False \ --num_nodes 1 \ --check_val_every_n_epoch 10 \ --finetune_from "$ckpt_path" \ data.params.batch_size="$BATCH_SIZE" \ lightning.trainer.accumulate_grad_batches="$ACCUMULATE_BATCHES" \ data.params.validation.params.n_gpus="$NUM_GPUS" \ ) Thanks.

[rorycochrane]

Getting the same issue. Also on Colab Pro.

Manually importing tf and keras didn't seem to help, neither did upgrading keras with pip.

[mercurialminutes]

same here!

[lvsi-qi]

Did you have solved it?

[aduchon]

I had the same issue with a Stable Diffusion upscaler. The solution there was replace !pip install einops==0.3.0 with !pip install einops==0.3.2

[nickwong64]

Thanks @aduchon. Your solution works!