Spaces:
Build error
Build error
Commit
·
3f437c8
1
Parent(s):
7a1b0a5
Update app.py
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ if torch.cuda.is_available():
|
|
| 36 |
variant='fp16')
|
| 37 |
|
| 38 |
# LoRA for 360° images
|
| 39 |
-
pipe.unet.load_attn_procs('artificialguybr/360Redmond')
|
| 40 |
|
| 41 |
if ENABLE_CPU_OFFLOAD:
|
| 42 |
pipe.enable_model_cpu_offload()
|
|
|
|
| 36 |
variant='fp16')
|
| 37 |
|
| 38 |
# LoRA for 360° images
|
| 39 |
+
pipe.unet.load_attn_procs('artificialguybr/360Redmond', weight_name='View360.safetensors')
|
| 40 |
|
| 41 |
if ENABLE_CPU_OFFLOAD:
|
| 42 |
pipe.enable_model_cpu_offload()
|