Spaces:
Running
on
Zero
Running
on
Zero
Update style_transfer.py
Browse files- style_transfer.py +0 -3
style_transfer.py
CHANGED
|
@@ -8,9 +8,6 @@ import os
|
|
| 8 |
import base64
|
| 9 |
from io import BytesIO
|
| 10 |
|
| 11 |
-
os.environ['TORCH_ENABLE_MPS_FALLBACK'] = '1'
|
| 12 |
-
os.environ['PYTHONWARNINGS'] = 'ignore::DeprecationWarning,ignore::UserWarning'
|
| 13 |
-
|
| 14 |
class DogStyleTransfer:
|
| 15 |
"""
|
| 16 |
Class for handling dog image style transfer using Stable Diffusion.
|
|
|
|
| 8 |
import base64
|
| 9 |
from io import BytesIO
|
| 10 |
|
|
|
|
|
|
|
|
|
|
| 11 |
class DogStyleTransfer:
|
| 12 |
"""
|
| 13 |
Class for handling dog image style transfer using Stable Diffusion.
|