Spaces:

CreitinGameplays
/

ConvAIChat

Paused

CreitinGameplays commited on Jun 1

Commit

5c2ab5b

•

1 Parent(s): fab78ee

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,9 +7,9 @@ import spaces
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
-MAX_MAX_NEW_TOKENS = 2048
 DEFAULT_MAX_NEW_TOKENS = 1024
-MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
 DESCRIPTION = """\
 # ConvAI 9b v2 Chat

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
+MAX_MAX_NEW_TOKENS = 4096
 DEFAULT_MAX_NEW_TOKENS = 1024
+MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "9000"))
 DESCRIPTION = """\
 # ConvAI 9b v2 Chat