Po II wojnie światowej cały majątek przeszedł w ręce państwowe – funkcjonowało tu państwowe gospodarstwo rolne, a zabudowania gorzelni stopniowo popadały w ruinę, aż do całkowitego wygaszenia produkcji pod koniec lat 90. XX wieku. Pałac natomiast, od 1963 roku, działa jako Wojewódzki Ośrodek Rehabilitacji w Kochcicach.
Prawdziwa przemiana Żółtego Folwarku nastąpiła w latach 2020–2024, kiedy to gmina Kochanowice, przy wsparciu funduszy unijnych, przeprowadziła gruntowną renowację dawnej gorzelni i ujeżdżalni. Powstało tu unikatowe muzeum techniki i kultury, w którym zachowano oryginalne elementy – ceglany komin, wieżę, dawne drzwi i nawet zabytkową maszynę parową.
Całość obiektu została dostosowana do potrzeb osób z niepełnosprawnościami – są windy, podjazdy, a także nowoczesna sala kinowa i przestrzenie ekspozycyjne. Z przeszklenia w wieży rozciąga się panoramiczny widok na dawny majątek Ballestremów.