michaelj commited on
Commit
4b9c74a
1 Parent(s): 6fd60b5

remove deepcache

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -15,12 +15,12 @@ adapter_id = "latent-consistency/lcm-lora-sdv1-5"
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
  pipe = AutoPipelineForImage2Image.from_pretrained("Lykon/dreamshaper-xl-v2-turbo", torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo")
17
  pipe = pipe.to(device)
18
- helper = DeepCacheSDHelper(pipe=pipe)
19
- helper.set_params(
20
- cache_interval=3,
21
- cache_branch_id=0,
22
- )
23
- helper.enable()
24
  def resize(value,img):
25
  img = Image.open(img)
26
  img = img.resize((value,value))
 
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
  pipe = AutoPipelineForImage2Image.from_pretrained("Lykon/dreamshaper-xl-v2-turbo", torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo")
17
  pipe = pipe.to(device)
18
+ # helper = DeepCacheSDHelper(pipe=pipe)
19
+ # helper.set_params(
20
+ # cache_interval=3,
21
+ # cache_branch_id=0,
22
+ # )
23
+ # helper.enable()
24
  def resize(value,img):
25
  img = Image.open(img)
26
  img = img.resize((value,value))