TheVeshup commited on
Commit
d3d6b34
·
verified ·
1 Parent(s): 2a2d2ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,10 +20,11 @@ else:
20
  torch_dtype = torch.float32
21
 
22
  pipe = DiffusionPipeline.from_pretrained(
23
- model_repo_id, torch_dtype=torch_dtype, use_auth_token=HF_TOKEN
24
  )
25
  pipe = pipe.to(device)
26
 
 
27
  MAX_SEED = np.iinfo(np.int32).max
28
  MAX_IMAGE_SIZE = 1024
29
 
 
20
  torch_dtype = torch.float32
21
 
22
  pipe = DiffusionPipeline.from_pretrained(
23
+ model_repo_id, torch_dtype=torch_dtype, token=HF_TOKEN
24
  )
25
  pipe = pipe.to(device)
26
 
27
+
28
  MAX_SEED = np.iinfo(np.int32).max
29
  MAX_IMAGE_SIZE = 1024
30