Hugging Face's logo

Spaces:
lochn
/
text_to_image
Sleeping

App Files Community
lochn commited on
Commit
0603590
·
verified ·
1 Parent(s): acd5678

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -14,8 +14,7 @@ if torch.cuda.is_available():
14
  else:
15
  torch_dtype = torch.float32
16
 
17
- pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
18
- pipe = pipe.to(device)
19
 
20
  MAX_SEED = np.iinfo(np.int32).max
21
  MAX_IMAGE_SIZE = 1024
 
14
  else:
15
  torch_dtype = torch.float32
16
 
17
+ pipe = DiffusionPipeline.from_pretrained(model_repo_id)
 
18
 
19
  MAX_SEED = np.iinfo(np.int32).max
20
  MAX_IMAGE_SIZE = 1024