changes in app.py
Browse files
app.py
CHANGED
|
@@ -17,6 +17,7 @@ model = "meta-llama/Llama-2-13b-chat-hf"
|
|
| 17 |
pipeline = transformers.pipeline(
|
| 18 |
"text-generation",
|
| 19 |
model=model,
|
|
|
|
| 20 |
device_map="auto",
|
| 21 |
do_sample=True,
|
| 22 |
token=HF_TOKEN,
|
|
|
|
| 17 |
pipeline = transformers.pipeline(
|
| 18 |
"text-generation",
|
| 19 |
model=model,
|
| 20 |
+
torch_dtype=torch.float32,
|
| 21 |
device_map="auto",
|
| 22 |
do_sample=True,
|
| 23 |
token=HF_TOKEN,
|