zeyadcode commited on
Commit
52e8406
·
verified ·
1 Parent(s): da5f013

Sync from GitHub via hub-sync

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -34,7 +34,7 @@ RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(
34
  RUN python -c "from ultralytics import SAM; import os; os.chdir('./models'); SAM('mobile_sam.pt')"
35
  RUN python -c "from transformers import AutoProcessor; AutoProcessor.from_pretrained('google/owlv2-base-patch16-ensemble')"
36
  RUN python -c "import torch; original_load = torch.jit.load; torch.jit.load = lambda *a, **kw: original_load(*a, **{**kw, 'map_location': 'cpu'}); from simple_lama_inpainting import SimpleLama; SimpleLama()"
37
- RUN python -c "from optimum.intel import OVStableDiffusionXLInpaintPipeline; pipe = OVStableDiffusionXLInpaintPipeline.from_pretrained('stabilityai/sdxl-turbo', export=True, device='CPU'); pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1); pipe.save_pretrained('./models/ov_sdxl_turbo_inpaint')"
38
 
39
 
40
  # Step 4: Copy the rest of the application
 
34
  RUN python -c "from ultralytics import SAM; import os; os.chdir('./models'); SAM('mobile_sam.pt')"
35
  RUN python -c "from transformers import AutoProcessor; AutoProcessor.from_pretrained('google/owlv2-base-patch16-ensemble')"
36
  RUN python -c "import torch; original_load = torch.jit.load; torch.jit.load = lambda *a, **kw: original_load(*a, **{**kw, 'map_location': 'cpu'}); from simple_lama_inpainting import SimpleLama; SimpleLama()"
37
+ RUN python -c "from huggingface_hub import snapshot_download; snapshot_download('zeyadcode/ov_sdxl_turbo_inpaint', local_dir='./models/ov_sdxl_turbo_inpaint')"
38
 
39
 
40
  # Step 4: Copy the rest of the application