CreitinGameplays commited on
Commit
04f5074
·
verified ·
1 Parent(s): 57ce3a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ DEFAULT_MAX_NEW_TOKENS = 1024
12
  MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "9000"))
13
 
14
  DESCRIPTION = """\
15
- # ConvAI 9b v2 Chat
16
  """
17
 
18
  if not torch.cuda.is_available():
@@ -20,7 +20,7 @@ if not torch.cuda.is_available():
20
 
21
 
22
  if torch.cuda.is_available():
23
- model_id = "CreitinGameplays/ConvAI-9b-v2"
24
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
25
  tokenizer = AutoTokenizer.from_pretrained(model_id)
26
  tokenizer.use_default_system_prompt = False
 
12
  MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "9000"))
13
 
14
  DESCRIPTION = """\
15
+ # MeowGPT 3.5 Chat
16
  """
17
 
18
  if not torch.cuda.is_available():
 
20
 
21
 
22
  if torch.cuda.is_available():
23
+ model_id = "cutycat2000x/MeowGPT-3.5"
24
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
25
  tokenizer = AutoTokenizer.from_pretrained(model_id)
26
  tokenizer.use_default_system_prompt = False