lfernandopg commited on
Commit
c5b8684
·
1 Parent(s): 8d58523

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -53,8 +53,9 @@ class StableDiffusionLoader:
53
 
54
  pipe = StableDiffusionPipeline.from_pretrained(
55
  self.pretrain_pipe,
56
- torch_dtype=torch.float16,
57
- )
 
58
  pipe = pipe.to(self.device)
59
  with autocast(self.device):
60
  image = pipe(self.prompt)[0][0]
 
53
 
54
  pipe = StableDiffusionPipeline.from_pretrained(
55
  self.pretrain_pipe,
56
+ #torch_dtype=torch.float16,
57
+ use_auth_token=False
58
+ ).to(device)
59
  pipe = pipe.to(self.device)
60
  with autocast(self.device):
61
  image = pipe(self.prompt)[0][0]