Cyril
/
ai-lab-transformers-playground
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

develop #3

Merged
Cyril merged 2 commits from develop into main 2025-10-18 15:13:39 +00:00
Conversation 0 Commits 2 Files Changed 6 +1 -1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 79227d6def - Show all commits

2
src/pipelines/textgen.py
View File

@ -30,7 +30,7 @@ class TextGenerator:
print("Model loaded successfully!") print("Model loaded successfully!")
def generate(self, prompt: str, max_new_tokens: int = 100, num_return_sequences: int = 1, def generate(self, prompt: str, max_new_tokens: int = 500, num_return_sequences: int = 1,
temperature: float = 1.0, do_sample: bool = True) -> Dict: temperature: float = 1.0, do_sample: bool = True) -> Dict:
""" """
Generate text from a prompt Generate text from a prompt