Spaces:

czl
/

generative-data-augmentation-demo

Running on Zero

czl commited on Jun 11

Commit

bceaa35

•

1 Parent(s): 3da643f

disable safetensors

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,14 +15,12 @@ if torch.cuda.is_available():
         model_path=model_path,
         device=device,
         use_torchcompile=False,
-        use_safetensors=True,
     )
 else:
     pipe = synth.pipe_img(
         model_path=model_path,
         device=device,
         use_torchcompile=False,
-        use_safetensors=True,
     )
 MAX_SEED = np.iinfo(np.int32).max

         model_path=model_path,
         device=device,
         use_torchcompile=False,
     )
 else:
     pipe = synth.pipe_img(
         model_path=model_path,
         device=device,
         use_torchcompile=False,
     )
 MAX_SEED = np.iinfo(np.int32).max