Edit on GitHub
  • John Snow Labs
  • Jan 24, 2025

English gpt2_medium_650k_stable_diffusion_prompt_generator_pipeline pipeline GPT2Transformer from Ar4ikov

  • en
  • open_source
  • pipeline
  • onnx

Description

Pretrained GPT2Transformer, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.gpt2_medium_650k_stable_diffusion_prompt_generator_pipeline is a English model originally trained by Ar4ikov.

Download Copy S3 URI

How to use


pipeline = PretrainedPipeline("gpt2_medium_650k_stable_diffusion_prompt_generator_pipeline", lang = "en")
annotations =  pipeline.transform(df)   



val pipeline = new PretrainedPipeline("gpt2_medium_650k_stable_diffusion_prompt_generator_pipeline", lang = "en")
val annotations = pipeline.transform(df)

Model Information

Model Name: gpt2_medium_650k_stable_diffusion_prompt_generator_pipeline
Type: pipeline
Compatibility: Spark NLP 5.5.1+
License: Open Source
Edition: Official
Language: en
Size: 1.3 GB

References

https://huggingface.co/Ar4ikov/gpt2-medium-650k-stable-diffusion-prompt-generator

Included Models

  • DocumentAssembler
  • GPT2Transformer