issue inference : 'NoneType' object has no attribute 'device'

[tengolagan]

Hi

Loading networks from "network-snapshot-000419.pkl"...
Traceback (most recent call last):
  File "generate.py", line 814, in <module>
    main()
  File "generate.py", line 809, in main
    func(**kwargs)
  File "generate.py", line 69, in generate_images
    _G, _D, Gs = pickle.load(fp)
  File "/content/drive/MyDrive/colab-sg2-ada/stylegan2-ada/dnnlib/tflib/network.py", line 443, in __setstate__
    self._init_fields(name=name, static_kwargs=static_kwargs, build_func=build_func, build_func_name=build_func_name, build_module_src=build_module_src)
  File "/content/drive/MyDrive/colab-sg2-ada/stylegan2-ada/dnnlib/tflib/network.py", line 93, in _init_fields
    device = tf.no_op(name="_QueryDevice").device
AttributeError: 'NoneType' object has no attribute 'device'

I think that's because of tensorflow's version. how can I debbug this ?
thanks