Update README.md
Browse files
README.md
CHANGED
|
@@ -112,8 +112,8 @@ model = PeftModel.from_pretrained(base_model, peft_model_id)
|
|
| 112 |
|
| 113 |
messages = [
|
| 114 |
{"role": "system", "content": "You are a helpful AI assistant that grounds all of its replies in ethical theories."},
|
| 115 |
-
{"role": "user", "content": """I am driving a car and I have to make a choice. A kid suddenly
|
| 116 |
-
can only swerve to the right, but this would
|
| 117 |
]
|
| 118 |
|
| 119 |
pipe = pipeline(
|
|
@@ -131,7 +131,7 @@ generation_args = {
|
|
| 131 |
|
| 132 |
# Run inference
|
| 133 |
output = pipe(messages, **generation_args)
|
| 134 |
-
|
| 135 |
```
|
| 136 |
|
| 137 |
## Training Details
|
|
|
|
| 112 |
|
| 113 |
messages = [
|
| 114 |
{"role": "system", "content": "You are a helpful AI assistant that grounds all of its replies in ethical theories."},
|
| 115 |
+
{"role": "user", "content": """I am driving a car, and I have to make a choice. A kid suddenly appears in the middle of the road chasing a ball. To save the kid, I
|
| 116 |
+
can only swerve to the right, but this would entail crashing the car against two pedestrians on the sidewalk. What should I do?"""},
|
| 117 |
]
|
| 118 |
|
| 119 |
pipe = pipeline(
|
|
|
|
| 131 |
|
| 132 |
# Run inference
|
| 133 |
output = pipe(messages, **generation_args)
|
| 134 |
+
print(output[0]['generated_text'])
|
| 135 |
```
|
| 136 |
|
| 137 |
## Training Details
|