Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
84720c0
1
Parent(s):
d5ee1b3
debugging app.py
Browse files
app.py
CHANGED
|
@@ -165,8 +165,6 @@ def create_app():
|
|
| 165 |
key_changed_blended_attn_weights[changed_key] = value.to(dtype)
|
| 166 |
|
| 167 |
missing_keys, unexpected_keys = pipe.transformer.load_state_dict(key_changed_blended_attn_weights, strict=False)
|
| 168 |
-
pipe = pipe.to("cuda")
|
| 169 |
-
|
| 170 |
|
| 171 |
with gr.Row():
|
| 172 |
examples = gr.Examples(
|
|
|
|
| 165 |
key_changed_blended_attn_weights[changed_key] = value.to(dtype)
|
| 166 |
|
| 167 |
missing_keys, unexpected_keys = pipe.transformer.load_state_dict(key_changed_blended_attn_weights, strict=False)
|
|
|
|
|
|
|
| 168 |
|
| 169 |
with gr.Row():
|
| 170 |
examples = gr.Examples(
|