recoilme commited on
Commit
fe22662
·
verified ·
1 Parent(s): 168372f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,6 +7,7 @@ from diffusers import DiffusionPipeline, AutoencoderKL, UNet2DConditionModel, Fl
7
  from transformers import AutoTokenizer, AutoModel
8
  from diffusers import DiffusionPipeline
9
 
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
  dtype = torch.float16 if torch.cuda.is_available() else torch.float32
12
  model_repo_id = "AiArtLab/sdxs"
@@ -14,7 +15,7 @@ model_repo_id = "AiArtLab/sdxs"
14
  pipe = DiffusionPipeline.from_pretrained(
15
  model_repo_id,
16
  torch_dtype=dtype,
17
- trust_remote_code=True
18
  ).to(device)
19
 
20
  MAX_SEED = np.iinfo(np.int32).max
 
7
  from transformers import AutoTokenizer, AutoModel
8
  from diffusers import DiffusionPipeline
9
 
10
+
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
  dtype = torch.float16 if torch.cuda.is_available() else torch.float32
13
  model_repo_id = "AiArtLab/sdxs"
 
15
  pipe = DiffusionPipeline.from_pretrained(
16
  model_repo_id,
17
  torch_dtype=dtype,
18
+ #trust_remote_code=True
19
  ).to(device)
20
 
21
  MAX_SEED = np.iinfo(np.int32).max