Update README.md
Browse files
README.md
CHANGED
@@ -45,4 +45,7 @@ pipe = pipeline('text2text-generation', model=model, tokenizer=tokenizer)
|
|
45 |
pipe(text)
|
46 |
#[{'generated_text': 'theore, major tech are clamouring to replicate the generative AI chatbot developed by OpenAI using its AI'}]
|
47 |
|
|
|
|
|
|
|
48 |
```
|
|
|
45 |
pipe(text)
|
46 |
#[{'generated_text': 'theore, major tech are clamouring to replicate the generative AI chatbot developed by OpenAI using its AI'}]
|
47 |
|
48 |
+
print("Model memory usage: {:.2f} MB".format(pipe.model.get_memory_footprint()/1e6))
|
49 |
+
# 'Model memory usage: 1893.99 MB'
|
50 |
+
|
51 |
```
|