Hugging Face's logo

Spaces:
rynmurdock
/
generative_recsys
Running on A10G

App Files Community
5
rynmurdock commited on
Commit
9a8a6a4
•
1 Parent(s): f8cc74c

.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ pipe.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-
49
  pipe.register_modules(image_encoder = image_encoder)
50
  pipe.set_ip_adapter_scale(0.8)
51
 
52
- pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesdxl", torch_dtype=torch.float16, low_cpu_mem_usage=True, device_map=DEVICE)
53
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
54
 
55
  pipe.to(device=DEVICE)
 
49
  pipe.register_modules(image_encoder = image_encoder)
50
  pipe.set_ip_adapter_scale(0.8)
51
 
52
+ pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taesdxl", torch_dtype=torch.float16, low_cpu_mem_usage=True)
53
  pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
54
 
55
  pipe.to(device=DEVICE)