alfredplpl commited on
Commit
33abad0
1 Parent(s): 1912a7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ weight_dtype_te = torch.bfloat16
14
  MAX_SEED = np.iinfo(np.int32).max
15
 
16
  transformer = Transformer2DModel.from_pretrained(
17
- "alfredplpl/commonart-tmp-3",
18
  torch_dtype=weight_dtype,
19
  token=os.environ["TOKEN"]
20
  )
 
14
  MAX_SEED = np.iinfo(np.int32).max
15
 
16
  transformer = Transformer2DModel.from_pretrained(
17
+ "aipicasso/commonart-beta",
18
  torch_dtype=weight_dtype,
19
  token=os.environ["TOKEN"]
20
  )