Spaces:
Running
Running
ayan4m1
commited on
Commit
·
3e69d5d
1
Parent(s):
f8e5f02
feat: add distilgpt2 model
Browse files
app.py
CHANGED
|
@@ -3,6 +3,7 @@ from transformers import pipeline
|
|
| 3 |
|
| 4 |
models = {
|
| 5 |
"MagicPrompt": "pszemraj/distilgpt2-magicprompt-SD",
|
|
|
|
| 6 |
"Llama-SmolTalk-3.2-1B": "prithivMLmods/Llama-SmolTalk-3.2-1B-Instruct"
|
| 7 |
}
|
| 8 |
pipelines = {}
|
|
|
|
| 3 |
|
| 4 |
models = {
|
| 5 |
"MagicPrompt": "pszemraj/distilgpt2-magicprompt-SD",
|
| 6 |
+
"DistilGPT2 SD": "FredZhang7/distilgpt2-stable-diffusion",
|
| 7 |
"Llama-SmolTalk-3.2-1B": "prithivMLmods/Llama-SmolTalk-3.2-1B-Instruct"
|
| 8 |
}
|
| 9 |
pipelines = {}
|