rynmurdock commited on
Commit
f47c034
1 Parent(s): 97656de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ start_time = time.time()
32
  ####################### Setup Model
33
  model_id = "stabilityai/stable-diffusion-xl-base-1.0"
34
  lcm_lora_id = "latent-consistency/lcm-lora-sdxl"
35
- pipe = SDEmb.from_pretrained(model_id, variant="fp16", low_mem_cpu_usage=True, revision="fp16")
36
  pipe.load_lora_weights(lcm_lora_id)
37
  pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
38
  pipe.to(device='cuda', dtype=torch.float16)
 
32
  ####################### Setup Model
33
  model_id = "stabilityai/stable-diffusion-xl-base-1.0"
34
  lcm_lora_id = "latent-consistency/lcm-lora-sdxl"
35
+ pipe = SDEmb.from_pretrained(model_id, variant="fp16", low_mem_cpu_usage=True)
36
  pipe.load_lora_weights(lcm_lora_id)
37
  pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
38
  pipe.to(device='cuda', dtype=torch.float16)