Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -43,7 +43,7 @@ class Item(BaseModel):
|
|
43 |
{"input": "Who do you love?", "output": "That’s a secret! But I care about all my friends. 😊"},
|
44 |
]
|
45 |
temperature: float = 0.7 # Controls randomness (0 = predictable, 1 = highly random)
|
46 |
-
max_new_tokens: int =
|
47 |
top_p: float = 0.9 # Sampling parameter for diverse responses
|
48 |
repetition_penalty: float = 1.1 # Prevents repetition
|
49 |
key: str = None # API key if needed
|
|
|
43 |
{"input": "Who do you love?", "output": "That’s a secret! But I care about all my friends. 😊"},
|
44 |
]
|
45 |
temperature: float = 0.7 # Controls randomness (0 = predictable, 1 = highly random)
|
46 |
+
max_new_tokens: int = 100 # Maximum response length
|
47 |
top_p: float = 0.9 # Sampling parameter for diverse responses
|
48 |
repetition_penalty: float = 1.1 # Prevents repetition
|
49 |
key: str = None # API key if needed
|