search

donahowe / TheaterGen

TheaterGen: Character Management with LLM for Consistent Multi-turn Image Generation
https://howe140.github.io/theatergen.io/
48 stars 2 forks source link

Missing Hugging Face Models and Docker Container Request #5

Closed Frank-Y-Zheng closed 1 month ago

Frank-Y-Zheng commented 1 month ago

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!