Probably just the US. But the US could do what EU has done with e.g. GDPR, Digital Services Act, USB-C regs, where they force any company trading in their region to follow those regulations for domestic customers.
And basically any AI company has to sell to US companies or consumers. That'd probs be sufficient to force them to use US models.