You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Unable to use flux fp8 model from Kijai/flux-fp8 while having transformer_flux.py file in local. I have modified the scripts to remove any import error. I put some print statements in single_model_file.py to check why it is not loading the model.
Reproduction
The below code works fine.
single_model_file.py
def _get_single_file_loadable_mapping_class(cls):
print(cls)
diffusers_module = importlib.import_module(__name__.split(".")[0])
for loadable_class_str in SINGLE_FILE_LOADABLE_CLASSES:
loadable_class = getattr(diffusers_module, loadable_class_str)
print(cls, loadable_class)
print(issubclass(cls, loadable_class))
if issubclass(cls, loadable_class):
return loadable_class_str
return None
from diffusers import FluxTransformer2DModel
transformer = FluxTransformer2DModel.from_single_file(
"https://huggingface.co/Kijai/flux-fp8/blob/main/flux1-schnell-fp8-e4m3fn.safetensors",
torch_dtype=torch.bfloat16
)
Edit: My bad, you can pass URLs to from_single_file, I got confused with something else. I think you just have to pass subfolder="transformer" when initializing to make your snippet work. This is because we try to fetch the init config from the original Flux-Dev repository (as Kijai/flux-fp8 is based on Flux-Dev).
Describe the bug
Unable to use flux fp8 model from
Kijai/flux-fp8
while having transformer_flux.py file in local. I have modified the scripts to remove any import error. I put some print statements in single_model_file.py to check why it is not loading the model.Reproduction
The below code works fine.
single_model_file.py
I am getting the below output:
But while using the class from my local code:
It is giving me following error:
Any leads would be appreciated.
Logs
No response
System Info
Who can help?
@DN6 @sayakpaul
The text was updated successfully, but these errors were encountered: