donahowe/TheaterGen

Missing Hugging Face Models and Docker Container Request

Closed this issue · 0 comments

When I run run.py, there are three missing Hugging Face models:

1.	/data2/chengjunhao/THEATERGEN/pretrained_models/vae_ft_mse
2.	/data2/chengjunhao/THEATERGEN/pretrained_models/diffusion_1.5_comic
3.	/diffusion_1.5/unet

Is it possible for you to provide the correct Hugging Face links for these models or their weights? Additionally, it would be greatly appreciated if you could provide a Docker container for this setup.

Thank you!