sachit-menon
/

illustrated_instructions

ShowNotTellPipeline

Model card Files Files and versions Community

sachit-menon commited on Jan 22

Commit

d46fb22

•

1 Parent(s): ef02a1a

Update snt_pipeline.py

Files changed (1) hide show

snt_pipeline.py +1 -1

snt_pipeline.py CHANGED Viewed

@@ -39,7 +39,7 @@ from diffusers.pipelines.pipeline_utils import DiffusionPipeline
 from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
 from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
-from trainer.models.sd_model import SDModel
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name

 from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
 from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
+from sd_model import SDModel
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name