Hugging Face's logo

Spaces:
Hodely
/
Amside
Sleeping

App Files Community
Hodely commited on
Commit
214e68c
·
verified ·
1 Parent(s): 021b62c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import AutoTokenizer, AutoModelForCausalLM
2
  import torch
3
 
4
- model_id = "mistralai/Mistral-7B-Instruct-v0.2"
5
  tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False)
6
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
7
 
 
1
  from transformers import AutoTokenizer, AutoModelForCausalLM
2
  import torch
3
 
4
+ model_id = "teknium/OpenHermes-2.5-Mistral-7B"
5
  tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=False)
6
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
7