sandelrt commited on
Commit
0fd6100
1 Parent(s): 4edf3ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -10,8 +10,7 @@ model_name = 'meta-llama/Meta-Llama-3-8B-Instruct'
10
 
11
  # Load the model
12
  model = transformers.pipeline("text-generation",
13
- model=model_name,
14
- model_kwargs={"torch_dtype": torch.float16})
15
 
16
  messages = []
17
 
 
10
 
11
  # Load the model
12
  model = transformers.pipeline("text-generation",
13
+ model=model_name)
 
14
 
15
  messages = []
16