Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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,
|