image-gpt
image-gpt copied to clipboard
How to run in windows or colab pro?
Do i need install linux, or it can be runned in windows env? Also, can it be run in 2x gtx 1080 ? And what averange speed of generation will be for one frame?
i try run in windows and get error: line 1801, in init self._traceback = tf_stack.extract_stack()
DataLossError (see above for traceback): Unable to open table file :\image-gpt\image-gpt-master\download: Unknown: NewRandomAccessFile failed to Create/Open: \image-gpt-master\download : Access is denied. ; Input/output error [[node save/RestoreV2 (defined at src/run.py:179) ]] [[node save/RestoreV2 (defined at src/run.py:179) ]]
I have the same issue
You must specify the name of the template, not the directory.
In my case :
python src/run.py --sample --n_embd 512 --n_head 8 --n_layer 24 --n_gpu 1 --ckpt_path checkpoints\model.ckpt-1000000 --data_path dataset\imagenet --color_cluster_path color\kmeans_centers.npy --save_dir out
Furthermore, if you get a memory error, you can add this to the beginning of the run.py
:
os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"
i-GPT2 work fine on single GPU (RTX2080TI) on Windows 10 (~3it/s).