Error in C1W1_3

I get following error when trying to generate the images in following cell:

fmt = dict(func=tflib.convert_images_to_uint8, nchw_to_nhwc=True)
images = Gs.run(noise_vectors, None, truncation_psi=Truncation, randomize_noise=False, output_transform=fmt)

Cannot assign a device for operation Gs/_Run/Gs/latents_in: {{node Gs/_Run/Gs/latents_in}} was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:XLA_CPU:0, /job:localhost/replica:0/task:0/device:XLA_GPU:0 ].

@Patrick, that’s the Pre-trained Model Exploration colab, right? Unfortunately, a couple of the colab optional labs are now out-of-date with the current colab environment and need to be updated. The developers just today temporarily hid these labs while they work on the update so students won’t run into exactly this problem.

You can watch this thread for more updates: Pre-trained Model Exploration