Diazepam prescribed

My wife takes Diazapam for anxiety when flying. We are going on a tour in China which will involve several internal flights. Will she be allowed to take this prescribed medication into China. We are away from home for 3 months and China is the first country we will visit and are staying for 2 weeks. She has 3 months supply. Will this be a problem at Customs. Anyone got any solutions??