File size: 258 Bytes
9fd1204
 
 
 
 
 
1
2
3
4
5
6
7
from finetrainers.models.flux import FluxModelSpecification


class DummyFluxModelSpecification(FluxModelSpecification):
    def __init__(self, **kwargs):
        super().__init__(pretrained_model_name_or_path="hf-internal-testing/tiny-flux-pipe", **kwargs)