Spaces:
Running
on
Zero
Running
on
Zero
alfredplpl
commited on
Commit
•
31eb163
1
Parent(s):
ea30826
Update app.py
Browse files
app.py
CHANGED
@@ -50,7 +50,6 @@ compel = Compel(tokenizer=[pipe.tokenizer, pipe.tokenizer_2] ,
|
|
50 |
text_encoder=[pipe.text_encoder, pipe.text_encoder_2],
|
51 |
returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES_NON_NORMALIZED,
|
52 |
requires_pooled=[False, True])
|
53 |
-
compel=compel.to("cuda")
|
54 |
|
55 |
def error_str(error, title="Error"):
|
56 |
return f"""#### {title}
|
|
|
50 |
text_encoder=[pipe.text_encoder, pipe.text_encoder_2],
|
51 |
returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES_NON_NORMALIZED,
|
52 |
requires_pooled=[False, True])
|
|
|
53 |
|
54 |
def error_str(error, title="Error"):
|
55 |
return f"""#### {title}
|