TimoTM commited on
Commit
dc6cb5d
·
verified ·
1 Parent(s): 2129939

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- result = self._pipeline(prompt, max_length=self._max_new_tokens, do_sample=True, temperature=self._temperature)
 
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