aitextgen icon indicating copy to clipboard operation
aitextgen copied to clipboard

TypeError: __init__() got an unexpected keyword argument 'checkpoint_callback'

Open kageraaron opened this issue 3 years ago • 6 comments
trafficstars

Running on Google Colab here: https://colab.research.google.com/drive/15qBZx5y9rdaQSyWpsreMDnTiZ5IlN0zD?usp=sharing#scrollTo=aeXshJM-Cuaf

First block of code throws this warning

ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
tensorflow 2.8.2+zzzcolab20220719082949 requires tensorboard<2.9,>=2.8, but you have tensorboard 2.9.1 which is incompatible.

And error is thrown at

ai.train(file_name,
         line_by_line=False,
         from_cache=False,
         num_steps=3000,
         generate_every=1000,
         save_every=1000,
         save_gdrive=False,
         learning_rate=1e-3,
         fp16=False,
         batch_size=1, 
         )
[/usr/local/lib/python3.7/dist-packages/pytorch_lightning/utilities/argparse.py](https://localhost:8080/#) in insert_env_defaults(self, *args, **kwargs)
    343 
    344         # all args were already moved to kwargs
--> 345         return fn(self, **kwargs)
    346 
    347     return cast(_T, insert_env_defaults)

TypeError: __init__() got an unexpected keyword argument 'checkpoint_callback'

kageraaron avatar Aug 02 '22 19:08 kageraaron

Just reporting same problem in a local install running the Shakespeare example on CPU on Linux. Just tried to upgrade tensorflow and tensorboard to see if there was something wrong with a dependency. Maybe something changed in their upgrade but I can't see it. Will keep digging.

ephemeralfuture avatar Aug 03 '22 14:08 ephemeralfuture

Just here to report the same problem as @kageraaron when attempting to run model training in Google Colab aitextgen demo.

ccbn9 avatar Aug 03 '22 15:08 ccbn9

Just here to report the same problem too! I'm yet again running it within Google Colab.

justanotherbyte avatar Aug 03 '22 16:08 justanotherbyte

Need to upgrade tensorflow and update deprecated methods as shown in pytorch lighning 1.7.0. See my pull request

tientr avatar Aug 04 '22 03:08 tientr

I merged #191 and pushed 0.6.0. Let me know if that works!

minimaxir avatar Aug 09 '22 04:08 minimaxir

I merged #191 and pushed 0.6.0. Let me know if that works!

it does. thank you.

when will models bigger than 124m/125m be finetunable on colab tho

Blazeolmo avatar Aug 19 '22 12:08 Blazeolmo