hysts HF staff commited on
Commit
791824f
1 Parent(s): c93c7f0

Add a sample model

Browse files
Files changed (1) hide show
  1. 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 = ['patrickvonplaten/lora_dreambooth_dog_example']
 
 
 
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):