Make flax from_pretrained work with local subfolder

[mishig25]

from diffusers import FlaxUNet2DConditionModel

model, params = FlaxUNet2DConditionModel.from_pretrained("/home/mishig/sd-v1-4-flax", subfolder="unet")

on main, this would raise an error saying that /home/mishig/sd-v1-4-flax/{FLAX_MODEL} does not exist because it was not respecting argument subfolder when checking locally

Pytorch version already handled it correctly:

diffusers/src/diffusers/modeling_utils.py

Lines 297 to 300 in 8685699

	elif subfolder is not None and os.path.isfile(
	os.path.join(pretrained_model_name_or_path, subfolder, WEIGHTS_NAME)
	):
	model_file = os.path.join(pretrained_model_name_or_path, subfolder, WEIGHTS_NAME)

[HuggingFaceDocBuilderDev]

The documentation is not available anymore as the PR was closed or merged.

[patil-suraj]

Very cool, thanks a lot for fixing this!