WebGL prevents training wikitext on Firefox

[JulienVig]

Training wikitext on Firefox raises the following error when first calling model.fitDataset:

Error: Requested texture size [10145x10146] greater than WebGL maximum on this browser / GPU [8192x8192].

According to tensorflow/tfjs#689 this is due to a "fundamental limitation of Firefox". The issue doesn't happen on Chrome.

Potential solution: display a message recommending switching web browsers when a user tries to train wikitext in Firefox

[JulienVig]

The error can be bypassed by decreasing requirements. E.g. Firefox supports gpt-nano with batch size 1 and context length 128.