Update app.py

app.py

@@ -12,8 +12,6 @@ except ModuleNotFoundError:
     subprocess.check_call(["python", '-m', 'pip', 'install', 'flash_attn'])
     print(f'El paquete ha sido instalado.')
 
-#tokenizer = AutoTokenizer.from_pretrained("togethercomputer/RedPajama-INCITE-Chat-3B-v1")
-#model = AutoModelForCausalLM.from_pretrained("togethercomputer/RedPajama-INCITE-Chat-3B-v1", torch_dtype=torch.float16)
 model = AutoModelForCausalLM.from_pretrained("mattshumer/mistral-8x7b-chat", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained("mattshumer/mistral-8x7b-chat")
 model = model.to('cuda:0')