wei commited on
Commit
49082b1
·
1 Parent(s): d5bc4bf

set sdk version

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. app.py +1 -1
README.md CHANGED
@@ -4,9 +4,9 @@ emoji: 🖼
4
  colorFrom: purple
5
  colorTo: red
6
  sdk: gradio
7
- sdk_version: 4.42.0
8
  app_file: app.py
9
  pinned: false
10
  ---
11
 
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
4
  colorFrom: purple
5
  colorTo: red
6
  sdk: gradio
7
+ sdk_version: 4.40.0
8
  app_file: app.py
9
  pinned: false
10
  ---
11
 
12
+ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
app.py CHANGED
@@ -14,7 +14,7 @@ else:
14
  torch_dtype = torch.float32
15
 
16
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch.bfloat16)
17
- #pipe.unet.load_attn_procs("enhanceaiteam/Flux-uncensored-V2")
18
  pipe = pipe.to(device)
19
 
20
  MAX_SEED = np.iinfo(np.int32).max
 
14
  torch_dtype = torch.float32
15
 
16
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch.bfloat16)
17
+ pipe.unet.load_attn_procs("enhanceaiteam/Flux-uncensored-V2")
18
  pipe = pipe.to(device)
19
 
20
  MAX_SEED = np.iinfo(np.int32).max