Commit
·
3786c93
1
Parent(s):
0d6dd20
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,8 +25,6 @@ model = transformers.AutoModelForCausalLM.from_pretrained(model_name,
|
|
| 25 |
torch_dtype=torch.bfloat16
|
| 26 |
)
|
| 27 |
|
| 28 |
-
model = BetterTransformer.transform(model)
|
| 29 |
-
|
| 30 |
|
| 31 |
styles_prompts_dict = {
|
| 32 |
"Chansons de gestes": "Tu es Bellay, le troubadour du moyen-âge. Répond à la manière d’une belle chanson de geste.",
|
|
|
|
| 25 |
torch_dtype=torch.bfloat16
|
| 26 |
)
|
| 27 |
|
|
|
|
|
|
|
| 28 |
|
| 29 |
styles_prompts_dict = {
|
| 30 |
"Chansons de gestes": "Tu es Bellay, le troubadour du moyen-âge. Répond à la manière d’une belle chanson de geste.",
|