Locutusque commited on
Commit
67d6e15
1 Parent(s): f6bf9f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import os
9
  # Install flash-attn
10
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
11
  # Initialize the model pipeline
12
- generator = pipeline('text-generation', model='Locutusque/OpenHercules-2.5-Mistral-7B', torch_dtype=torch.bfloat16, token=os.environ["HF"])
13
  @spaces.GPU
14
  def generate_text(prompt, temperature, top_p, top_k, repetition_penalty, max_length):
15
  # Generate text using the model
 
9
  # Install flash-attn
10
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
11
  # Initialize the model pipeline
12
+ generator = pipeline('text-generation', model='Locutusque/OpenHercules-2.5-Mistral-7B', torch_dtype=torch.bfloat16)
13
  @spaces.GPU
14
  def generate_text(prompt, temperature, top_p, top_k, repetition_penalty, max_length):
15
  # Generate text using the model