Ketansomewhere
/

gpt-neo

Text Generation

Generated from Trainer

text-generation-inference

Model card Files Files and versions

Metrics Training metrics Community

Ketansomewhere commited on Jul 23, 2024

Commit

41ec0e6

·

verified ·

1 Parent(s): b92c107

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -73,7 +73,7 @@ device = 'cuda'
 device = 0 if torch.cuda.is_available() else -1  # Use GPU if available, otherwise CPU
 # Create the pipeline
-text_generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=device)
 ```

 device = 0 if torch.cuda.is_available() else -1  # Use GPU if available, otherwise CPU
 # Create the pipeline
+text_generator = pipeline("text-generation", model="Ketansomewhere/gpt-neo", tokenizer=tokenizer, device=device)
 ```