aitextgen
aitextgen copied to clipboard
TypeError: __init__() got an unexpected keyword argument 'checkpoint_callback'
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'
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.
Just here to report the same problem as @kageraaron when attempting to run model training in Google Colab aitextgen demo.
Just here to report the same problem too! I'm yet again running it within Google Colab.
Need to upgrade tensorflow and update deprecated methods as shown in pytorch lighning 1.7.0. See my pull request
I merged #191 and pushed 0.6.0. Let me know if that works!
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