Hugging Face's logo

Spaces:
czl
/
generative-data-augmentation-demo
Running on Zero

App Files Community
1
czl commited on
Commit
bceaa35
·
verified ·
1 Parent(s): 3da643f

disable safetensors

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -15,14 +15,12 @@ if torch.cuda.is_available():
15
  model_path=model_path,
16
  device=device,
17
  use_torchcompile=False,
18
- use_safetensors=True,
19
  )
20
  else:
21
  pipe = synth.pipe_img(
22
  model_path=model_path,
23
  device=device,
24
  use_torchcompile=False,
25
- use_safetensors=True,
26
  )
27
 
28
  MAX_SEED = np.iinfo(np.int32).max
 
15
  model_path=model_path,
16
  device=device,
17
  use_torchcompile=False,
 
18
  )
19
  else:
20
  pipe = synth.pipe_img(
21
  model_path=model_path,
22
  device=device,
23
  use_torchcompile=False,
 
24
  )
25
 
26
  MAX_SEED = np.iinfo(np.int32).max