nickdigger commited on
Commit
b9251f1
·
verified ·
1 Parent(s): a53c16e

Fix torch_compile parameter error

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -43,8 +43,7 @@ def caption_image_optimized(image, style, length):
43
  MODEL_PATH,
44
  torch_dtype=torch.bfloat16,
45
  device_map="auto",
46
- low_cpu_mem_usage=True,
47
- torch_compile=False # Disable compilation for faster loading
48
  )
49
  model.eval()
50
 
 
43
  MODEL_PATH,
44
  torch_dtype=torch.bfloat16,
45
  device_map="auto",
46
+ low_cpu_mem_usage=True
 
47
  )
48
  model.eval()
49