ford442 commited on
Commit
71fee0a
·
verified ·
1 Parent(s): a8460c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -184,7 +184,7 @@ hidet.option.cache_dir("local_cache")
184
  #hidet.torch.dynamo_config.use_fp16_reduction(True)
185
  # use tensorcore
186
  hidet.torch.dynamo_config.use_tensor_core()
187
- hidet.torch.dynamo_config.steal_weights(False)
188
 
189
  pipe = load_and_prepare_model()
190
 
 
184
  #hidet.torch.dynamo_config.use_fp16_reduction(True)
185
  # use tensorcore
186
  hidet.torch.dynamo_config.use_tensor_core()
187
+ #hidet.torch.dynamo_config.steal_weights(False)
188
 
189
  pipe = load_and_prepare_model()
190