inoculatemedia commited on
Commit
7f8554d
·
verified ·
1 Parent(s): 6f27b23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -14,11 +14,9 @@ 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
- pipe = SemanticStableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
20
- pipe = pipe.to("cuda")
21
 
 
 
22
  out = pipe(
23
  prompt="a photo of the face of a woman",
24
  num_images_per_prompt=1,
 
14
  else:
15
  torch_dtype = torch.float32
16
 
 
 
 
 
17
 
18
+ pipe = SemanticStableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
19
+ pipe = pipe.to(device)
20
  out = pipe(
21
  prompt="a photo of the face of a woman",
22
  num_images_per_prompt=1,