Предпочтительно оформить так. В голове пронеслось: только бы не упасть...
Такой оборот вполне корректен: Вы, Николай Аполлонович, с своею инвалидною бородой были бы здесь невозможны: вам, как только бы вы вышли на улицу, непременно подадут милостыню [И. А. Гончаров. Фрегат «Паллада» (1855)]; Простого, понятного, русского языка Бальмонт боится, и совершенно резонно, ибо, как только бы он заговорил просто, сейчас же обнаружилось бы его убожество [К. Д. Бальмонт. Полное собрание стихов (16.04.1913) // «Московские ведомости», 1913] и т. п. Однако нужно заметить, что в нем имеется некоторая инверсия: частица бы более уместна после глагола, к которому относится (как только вы вышли бы, как только он заговорил бы).