Spaces:

benjamin-paine
/

nvidia-cosmos-prompt-upsampler

Running on Zero

benjamin-paine commited on 6 days ago

Commit

b7d72bd

verified ·

1 Parent(s): 08b845b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,6 +13,7 @@ llama = None
 @spaces.GPU
 def upsample(prompt):
     if llama is None:
         llama = Llama(
             model,

 @spaces.GPU
 def upsample(prompt):
+    global llama
     if llama is None:
         llama = Llama(
             model,