Add a sample model
Browse files- app_inference.py +4 -1
app_inference.py
CHANGED
|
@@ -10,7 +10,10 @@ from huggingface_hub import HfApi
|
|
| 10 |
from inference import InferencePipeline
|
| 11 |
from utils import find_exp_dirs
|
| 12 |
|
| 13 |
-
SAMPLE_MODEL_IDS = [
|
|
|
|
|
|
|
|
|
|
| 14 |
|
| 15 |
|
| 16 |
class ModelSource(enum.Enum):
|
|
|
|
| 10 |
from inference import InferencePipeline
|
| 11 |
from utils import find_exp_dirs
|
| 12 |
|
| 13 |
+
SAMPLE_MODEL_IDS = [
|
| 14 |
+
'patrickvonplaten/lora_dreambooth_dog_example',
|
| 15 |
+
'sayakpaul/sd-model-finetuned-lora-t4',
|
| 16 |
+
]
|
| 17 |
|
| 18 |
|
| 19 |
class ModelSource(enum.Enum):
|