dome272 commited on
Commit
f2d587a
1 Parent(s): 4da08fe

activate torch.compile

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ if not torch.cuda.is_available():
17
  MAX_SEED = np.iinfo(np.int32).max
18
  CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES") == "1"
19
  MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1536"))
20
- USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
21
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
22
  PREVIEW_IMAGES = True
23
 
 
17
  MAX_SEED = np.iinfo(np.int32).max
18
  CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES") == "1"
19
  MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1536"))
20
+ USE_TORCH_COMPILE = True
21
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
22
  PREVIEW_IMAGES = True
23