Update README.md
Browse files
README.md
CHANGED
@@ -54,12 +54,16 @@ device = "cuda" # for GPU usage or "cpu" for CPU usage
|
|
54 |
|
55 |
tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/cosmo-1b")
|
56 |
model = AutoModelForCausalLM.from_pretrained("HuggingFaceTB/cosmo-1b").to(device)
|
57 |
-
prompt = "
|
58 |
|
59 |
inputs = tokenizer(prompt, return_tensors="pt").to(device)
|
60 |
output = model.generate(**inputs, max_length=300, do_sample=True, temperature=0.6, top_p=0.95, repetition_penalty=1.2)
|
61 |
print(tokenizer.decode(output[0]))
|
62 |
```
|
|
|
|
|
|
|
|
|
63 |
|
64 |
# Limitations
|
65 |
|
|
|
54 |
|
55 |
tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/cosmo-1b")
|
56 |
model = AutoModelForCausalLM.from_pretrained("HuggingFaceTB/cosmo-1b").to(device)
|
57 |
+
prompt = "Photosynthesis is"
|
58 |
|
59 |
inputs = tokenizer(prompt, return_tensors="pt").to(device)
|
60 |
output = model.generate(**inputs, max_length=300, do_sample=True, temperature=0.6, top_p=0.95, repetition_penalty=1.2)
|
61 |
print(tokenizer.decode(output[0]))
|
62 |
```
|
63 |
+
Output:
|
64 |
+
```
|
65 |
+
<s> Photosynthesis is the process by which green plants, algae and some bacteria convert light energy into chemical energy in order to fuel their metabolic processes. The reaction takes place within specialized cells called chloroplasts. This article focuses on the electron transport chain (ETC), a critical part of photosystem II where most of the solar-driven electrons are passed through before being reduced to water.
|
66 |
+
```
|
67 |
|
68 |
# Limitations
|
69 |
|