Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = '
|
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)
|