patrickbdevaney commited on
Commit
06215ca
·
verified ·
1 Parent(s): 99d83b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_ite
94
  iteration_count = 0
95
 
96
  print("Initializing the text generation pipeline with 16-bit precision...")
97
- model_name = 'mlabonne/Meta-Llama-3.1-8B-Instruct-abliterated'
98
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map='auto')
99
  tokenizer = AutoTokenizer.from_pretrained(model_name)
100
  text_generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
 
94
  iteration_count = 0
95
 
96
  print("Initializing the text generation pipeline with 16-bit precision...")
97
+ model_name = 'NousResearch/Meta-Llama-3.1-8B-Instruct'
98
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map='auto')
99
  tokenizer = AutoTokenizer.from_pretrained(model_name)
100
  text_generator = pipeline('text-generation', model=model, tokenizer=tokenizer)