Spaces:

Karzan
/

google-gemma-2-9b

Runtime error

Karzan commited on Jun 28

Commit

f34d089

•

1 Parent(s): 5d03833

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ model = GemmaForCausalLM.from_pretrained('google/gemma-2-9b',
 # pipe = pipeline('text-generation', model=model,tokenizer = tokenizer)
 MAX_MAX_NEW_TOKENS = 2048
-DEFAULT_MAX_NEW_TOKENS = 128
 @spaces.GPU(duration=120)
 def generate(

 # pipe = pipeline('text-generation', model=model,tokenizer = tokenizer)
 MAX_MAX_NEW_TOKENS = 2048
+DEFAULT_MAX_NEW_TOKENS = 1024
 @spaces.GPU(duration=120)
 def generate(