dg845 commited on
Commit
df30d9a
1 Parent(s): ec19bf1

Update app.py

Browse files

Change model_ids for UniDiffuser checkpoints

Files changed (1) hide show
  1. app.py +2 -2
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 = "dg845/unidiffuser-diffusers"
10
- # model_id = "dg845/unidiffuser-diffusers-v0"
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
  )