Abhaykoul commited on
Commit
08d1030
·
verified ·
1 Parent(s): 41b8b62

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 = [