Hatman commited on
Commit
87fc56c
·
verified ·
1 Parent(s): 307c114

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -15,6 +15,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
15
  return seed
16
 
17
  device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 
18
  pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16).to(device)
19
  pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
20
 
@@ -55,6 +56,7 @@ def create_image(image_pil,
55
  print('image loaded')
56
  generator = torch.Generator().manual_seed(randomize_seed_fn(seed, False))
57
  print('generator loaded')
 
58
  image = pipeline(
59
  prompt=prompt,
60
  ip_adapter_image=style_image,
 
15
  return seed
16
 
17
  device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
18
+ print(torch.cuda.is_available())
19
  pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16).to(device)
20
  pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
21
 
 
56
  print('image loaded')
57
  generator = torch.Generator().manual_seed(randomize_seed_fn(seed, False))
58
  print('generator loaded')
59
+ print(pipeline.model.device)
60
  image = pipeline(
61
  prompt=prompt,
62
  ip_adapter_image=style_image,