darshan8950 commited on
Commit
5930792
·
verified ·
1 Parent(s): 7c962e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def main():
24
  'delimiter': ','})
25
  data = loader.load()
26
 
27
- model = "kittn/mistral-7B-v0.1-hf"
28
  tokenizer = AutoTokenizer.from_pretrained(model)
29
  pipeline = transformers.pipeline("text-generation", model=model, tokenizer=tokenizer, torch_dtype=torch.bfloat16, device_map="auto", do_sample=True, top_k=1, num_return_sequences=1, eos_token_id=tokenizer.eos_token_id,offload_folder="offload")
30
 
 
24
  'delimiter': ','})
25
  data = loader.load()
26
 
27
+ model = "stabilityai/stablelm-zephyr-3b"
28
  tokenizer = AutoTokenizer.from_pretrained(model)
29
  pipeline = transformers.pipeline("text-generation", model=model, tokenizer=tokenizer, torch_dtype=torch.bfloat16, device_map="auto", do_sample=True, top_k=1, num_return_sequences=1, eos_token_id=tokenizer.eos_token_id,offload_folder="offload")
30