Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,8 @@ class GermanGPT2(LLM):
|
|
| 21 |
self._pipeline = pipeline("text-generation", model="dbmdz/german-gpt2")
|
| 22 |
|
| 23 |
def _call(self, prompt, stop=None):
|
| 24 |
-
|
|
|
|
| 25 |
return result[0]["generated_text"]
|
| 26 |
|
| 27 |
@property
|
|
|
|
| 21 |
self._pipeline = pipeline("text-generation", model="dbmdz/german-gpt2")
|
| 22 |
|
| 23 |
def _call(self, prompt, stop=None):
|
| 24 |
+
# Nutze nun max_new_tokens anstatt max_length
|
| 25 |
+
result = self._pipeline(prompt, max_new_tokens=self._max_new_tokens, do_sample=True, temperature=self._temperature)
|
| 26 |
return result[0]["generated_text"]
|
| 27 |
|
| 28 |
@property
|