djai.model.models.ml.hugging_face.text_generation module

DjAI Pre-Trained Hugging Face Text Generator Model class.

class djai.model.models.ml.hugging_face.text_generation.PreTrainedHuggingFaceTextGenerator(*args, **kwargs)

Bases: djai.model.models.ml.hugging_face.base.PreTrainedHuggingFaceTransformer

DjAI Pre-Trained Hugging Face Text Generator Model class.

exception DoesNotExist

Bases: djai.model.models.ml.hugging_face.base.PreTrainedHuggingFaceTransformer.DoesNotExist

exception MultipleObjectsReturned

Bases: djai.model.models.ml.hugging_face.base.PreTrainedHuggingFaceTransformer.MultipleObjectsReturned

gradio_ui = Gradio Interface for: predict ----------------------------- inputs: |-Textbox(label="Text to Generate from") |-Checkbox(label="Return Tensors?") |-Checkbox(label="Return Text?") |-Checkbox(label="Return Full Text?") |-Checkbox(label="Clean Up Spaces?") |-Textbox(label="Generated Text Prefix") outputs: |-JSON(label="Generated Text")

polymorphic_primary_key_name = 'uuid'

polymorphic_super_sub_accessors_replaced = False

predict(text_or_texts: Union[str, collections.abc.Sequence[str]], return_tensors: bool = False, return_text: bool = True, return_full_text: bool = True, clean_up_tokenization_spaces: bool = True, prefix: Optional[str] = None, **generate_kwargs) → Union[str, List[int], List[Union[str, List[int]]]]

Generate Text(s).

pretrainedhuggingfacetransformer_ptr

Accessor to the related object on the forward side of a one-to-one relation.

In the example:

class Restaurant(Model):
    place = OneToOneField(Place, related_name='restaurant')

Restaurant.place is a ForwardOneToOneDescriptor instance.

pretrainedhuggingfacetransformer_ptr_id