Spaces:
Running
on
Zero
Running
on
Zero
Kunpeng Song
commited on
Commit
·
7c69fc1
1
Parent(s):
a6600be
fix zero
Browse files
.DS_Store
CHANGED
|
Binary files a/.DS_Store and b/.DS_Store differ
|
|
|
app.py
CHANGED
|
@@ -6,7 +6,7 @@ import numpy as np
|
|
| 6 |
import torch
|
| 7 |
from pytorch_lightning import seed_everything
|
| 8 |
from model_lib.utils import parse_args
|
| 9 |
-
from llava.mm_utils import process_image
|
| 10 |
|
| 11 |
os.environ["CUDA_VISIBLE_DEVICES"]="0"
|
| 12 |
|
|
@@ -19,7 +19,8 @@ args = parse_args()
|
|
| 19 |
model = None
|
| 20 |
|
| 21 |
def my_process_image(a, b, c):
|
| 22 |
-
return process_image(a, b, c)
|
|
|
|
| 23 |
|
| 24 |
@spaces.GPU
|
| 25 |
def inference(rgb, subject, prompt, strength, seed):
|
|
|
|
| 6 |
import torch
|
| 7 |
from pytorch_lightning import seed_everything
|
| 8 |
from model_lib.utils import parse_args
|
| 9 |
+
# from llava.mm_utils import process_image
|
| 10 |
|
| 11 |
os.environ["CUDA_VISIBLE_DEVICES"]="0"
|
| 12 |
|
|
|
|
| 19 |
model = None
|
| 20 |
|
| 21 |
def my_process_image(a, b, c):
|
| 22 |
+
# return process_image(a, b, c)
|
| 23 |
+
return (a, b, c)
|
| 24 |
|
| 25 |
@spaces.GPU
|
| 26 |
def inference(rgb, subject, prompt, strength, seed):
|