Update README.md
Browse files
README.md
CHANGED
|
@@ -61,7 +61,7 @@ from accelerate import Accelerator
|
|
| 61 |
accelerator = Accelerator()
|
| 62 |
|
| 63 |
# Initialize the pipeline
|
| 64 |
-
pipe = pipeline("text-generation", model="OEvortex/HelpingAI-Lite", device=accelerator.device)
|
| 65 |
|
| 66 |
# Define the messages
|
| 67 |
messages = [
|
|
|
|
| 61 |
accelerator = Accelerator()
|
| 62 |
|
| 63 |
# Initialize the pipeline
|
| 64 |
+
pipe = pipeline("text-generation", model="OEvortex/HelpingAI-Lite-2x1B", device=accelerator.device)
|
| 65 |
|
| 66 |
# Define the messages
|
| 67 |
messages = [
|