Update app.py
Browse filesChange model_ids for UniDiffuser checkpoints
app.py
CHANGED
@@ -6,8 +6,8 @@ from diffusers import UniDiffuserPipeline
|
|
6 |
|
7 |
|
8 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
9 |
-
model_id = "
|
10 |
-
# model_id = "
|
11 |
pipeline = UniDiffuserPipeline.from_pretrained(
|
12 |
model_id,
|
13 |
)
|
|
|
6 |
|
7 |
|
8 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
9 |
+
model_id = "thu-ml/unidiffuser-v1"
|
10 |
+
# model_id = "thu-ml/unidiffuser-v0"
|
11 |
pipeline = UniDiffuserPipeline.from_pretrained(
|
12 |
model_id,
|
13 |
)
|