eolecvk commited on
Commit
a5bc9e2
1 Parent(s): 9fc7101

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
9
  context = autocast if device == "cuda" else nullcontext
10
  dtype = torch.float16 if device == "cuda" else torch.float32
11
 
12
- model_id = 'eolecvk/dreambooth-avatar'
13
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=dtype)
14
  pipe = pipe.to(device)
15
 
 
9
  context = autocast if device == "cuda" else nullcontext
10
  dtype = torch.float16 if device == "cuda" else torch.float32
11
 
12
+ model_id = 'lambdalabs/dreambooth-avatar'
13
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=dtype)
14
  pipe = pipe.to(device)
15