Hugging Face's logo

Spaces:
Karzan
/
google-gemma-2-9b
Running on Zero

App Files Community
Karzan commited on
Commit
f34d089
•
1 Parent(s): 5d03833

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ model = GemmaForCausalLM.from_pretrained('google/gemma-2-9b',
11
  # pipe = pipeline('text-generation', model=model,tokenizer = tokenizer)
12
 
13
  MAX_MAX_NEW_TOKENS = 2048
14
- DEFAULT_MAX_NEW_TOKENS = 128
15
 
16
  @spaces.GPU(duration=120)
17
  def generate(
 
11
  # pipe = pipeline('text-generation', model=model,tokenizer = tokenizer)
12
 
13
  MAX_MAX_NEW_TOKENS = 2048
14
+ DEFAULT_MAX_NEW_TOKENS = 1024
15
 
16
  @spaces.GPU(duration=120)
17
  def generate(