Hugging Face's logo

Spaces:
aipicasso
/
commonart-latest
Running on Zero

App Files Community
alfredplpl commited on
Commit
a111da9
·
verified ·
1 Parent(s): 4238a64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from diffusers import Transformer2DModel, PixArtSigmaPipeline, AutoencoderKL, DP
9
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, QuantoConfig, EetqConfig
10
 
11
  device = "cuda"
12
- weight_dtype = torch.float32
13
  weight_dtype_te = torch.bfloat16
14
  MAX_SEED = np.iinfo(np.int32).max
15
 
 
9
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, QuantoConfig, EetqConfig
10
 
11
  device = "cuda"
12
+ weight_dtype = torch.bfloat16
13
  weight_dtype_te = torch.bfloat16
14
  MAX_SEED = np.iinfo(np.int32).max
15