Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -201,6 +201,7 @@ def inference(
|
|
201 |
text_encoder.resize_token_embeddings(len(tokenizer))
|
202 |
text_encoder.get_input_embeddings().weight.data[-delta_weight.shape[0]:] = delta_weight
|
203 |
|
|
|
204 |
config = {
|
205 |
"negative_prompt": neg_prompt,
|
206 |
"num_inference_steps": int(steps),
|
|
|
201 |
text_encoder.resize_token_embeddings(len(tokenizer))
|
202 |
text_encoder.get_input_embeddings().weight.data[-delta_weight.shape[0]:] = delta_weight
|
203 |
|
204 |
+
|
205 |
config = {
|
206 |
"negative_prompt": neg_prompt,
|
207 |
"num_inference_steps": int(steps),
|