JMalott commited on
Commit
7c67c9f
1 Parent(s): ab6e890

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -107,7 +107,7 @@ def generate2(prompt,crazy,k):
107
  models_root='./pretrained',
108
  dtype=torch.float32,
109
  device='cpu',
110
- is_mega=True,
111
  is_reusable=True
112
  )
113
 
@@ -119,7 +119,7 @@ def generate2(prompt,crazy,k):
119
  is_seamless=False,
120
  temperature=crazy,
121
  top_k=k,#2128,
122
- supercondition_factor=32,
123
  is_verbose=False
124
  )
125
 
 
107
  models_root='./pretrained',
108
  dtype=torch.float32,
109
  device='cpu',
110
+ is_mega=False,
111
  is_reusable=True
112
  )
113
 
 
119
  is_seamless=False,
120
  temperature=crazy,
121
  top_k=k,#2128,
122
+ supercondition_factor=64,
123
  is_verbose=False
124
  )
125