Update test_4bit.py
Browse files- test_4bit.py +1 -1
test_4bit.py
CHANGED
|
@@ -207,7 +207,7 @@ class changeClothes():
|
|
| 207 |
print("Load to CUDA ...........")
|
| 208 |
pipe.to("cuda")
|
| 209 |
|
| 210 |
-
|
| 211 |
|
| 212 |
@staticmethod
|
| 213 |
def check_image(image_input):
|
|
|
|
| 207 |
print("Load to CUDA ...........")
|
| 208 |
pipe.to("cuda")
|
| 209 |
|
| 210 |
+
return pipe
|
| 211 |
|
| 212 |
@staticmethod
|
| 213 |
def check_image(image_input):
|