Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
Β·
1d40feb
1
Parent(s):
8828289
Update app.py
Browse files
app.py
CHANGED
@@ -198,8 +198,8 @@ device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
|
198 |
print('Using device:', device)
|
199 |
model = load_vqgan_model(args.vqgan_config, args.vqgan_checkpoint).to(device)
|
200 |
perceptor = clip.load(args.clip_model, jit=False)[0].eval().requires_grad_(False).to(device)
|
201 |
-
all_frames = []
|
202 |
def inference(text, seed, step_size, max_iterations, width, height, init_image, init_weight):
|
|
|
203 |
size=[width, height]
|
204 |
texts = text
|
205 |
init_weight=init_weight
|
|
|
198 |
print('Using device:', device)
|
199 |
model = load_vqgan_model(args.vqgan_config, args.vqgan_checkpoint).to(device)
|
200 |
perceptor = clip.load(args.clip_model, jit=False)[0].eval().requires_grad_(False).to(device)
|
|
|
201 |
def inference(text, seed, step_size, max_iterations, width, height, init_image, init_weight):
|
202 |
+
all_frames = []
|
203 |
size=[width, height]
|
204 |
texts = text
|
205 |
init_weight=init_weight
|