Update README.md
Browse files
README.md
CHANGED
|
@@ -48,7 +48,6 @@ model_id = "suayptalha/FastLlama-3.2-1B-Instruct"
|
|
| 48 |
pipe = pipeline(
|
| 49 |
"text-generation",
|
| 50 |
model=model_id,
|
| 51 |
-
torch_dtype=torch.bfloat16,
|
| 52 |
device_map="auto",
|
| 53 |
)
|
| 54 |
messages = [
|
|
|
|
| 48 |
pipe = pipeline(
|
| 49 |
"text-generation",
|
| 50 |
model=model_id,
|
|
|
|
| 51 |
device_map="auto",
|
| 52 |
)
|
| 53 |
messages = [
|